Auto-commit: OCR workflow improvements, performance optimizations, and bug fixes

LightRAG-main/run_selenium_test.py

@@ -0,0 +1,75 @@
+#!/usr/bin/env python3
+"""
+Run Selenium test with a temporary LightRAG server.
+"""
+
+import subprocess
+import time
+import sys
+import os
+import urllib.request
+import urllib.error
+
+def is_server_running(url='http://localhost:8000', timeout=2):
+    try:
+        response = urllib.request.urlopen(url, timeout=timeout)
+        return response.status < 500
+    except:
+        return False
+
+def start_server():
+    """Start LightRAG server as a subprocess."""
+    # Change to LightRAG-main directory
+    cwd = os.path.dirname(os.path.abspath(__file__))
+    env = os.environ.copy()
+    # Ensure Python path includes current directory
+    env['PYTHONPATH'] = cwd + (os.pathsep + env.get('PYTHONPATH', ''))
+    cmd = [sys.executable, 'lightrag_server.py']
+    proc = subprocess.Popen(
+        cmd,
+        cwd=cwd,
+        env=env,
+        stdout=subprocess.PIPE,
+        stderr=subprocess.PIPE,
+        text=True
+    )
+    return proc
+
+def main():
+    print("Starting LightRAG server...")
+    server_proc = start_server()
+    
+    # Wait for server to be ready
+    max_wait = 30
+    start_time = time.time()
+    while time.time() - start_time < max_wait:
+        if is_server_running():
+            print("Server is ready.")
+            break
+        time.sleep(1)
+    else:
+        print("Server failed to start within 30 seconds.")
+        server_proc.terminate()
+        server_proc.wait()
+        sys.exit(1)
+    
+    # Run Selenium test
+    print("Running Selenium test...")
+    from test_workspace_ui_isolation import test_workspace_ui_isolation
+    try:
+        test_workspace_ui_isolation()
+        print("Selenium test completed successfully.")
+    except Exception as e:
+        print(f"Selenium test failed: {e}")
+        server_proc.terminate()
+        server_proc.wait()
+        sys.exit(1)
+    
+    # Stop server
+    print("Stopping server...")
+    server_proc.terminate()
+    server_proc.wait()
+    print("Done.")
+
+if __name__ == "__main__":
+    main()