dsfvgb

app.py

@@ -550,10 +550,20 @@ with gr.Blocks(title="Text-to-SQL RLHF") as demo:
 
 if __name__ == "__main__":
     server_name = os.environ.get("GRADIO_SERVER_NAME", "0.0.0.0")
-    server_port = 7860
+    base_port = int(os.environ.get("GRADIO_SERVER_PORT", 7860))
+    max_retries = 10
     
-    print(f"Starting Gradio UI on {server_name}:{server_port}...", flush=True)
-    try:
-        demo.launch(server_name=server_name, server_port=server_port, ssr_mode=False)
-    except TypeError:
-        demo.launch(server_name=server_name, server_port=server_port)
+    for port in range(base_port, base_port + max_retries):
+        try:
+            print(f"Attempting to start Gradio UI on {server_name}:{port}...", flush=True)
+            demo.launch(server_name=server_name, server_port=port)
+            break  # If successful, exit the loop
+        except OSError as e:
+            if "Cannot find empty port" in str(e) or "Address already in use" in str(e):
+                print(f"⚠️ Port {port} is in use, trying next port...")
+                continue
+            else:
+                # If it's a different OSError, raise it normally
+                raise e
+    else:
+        print(f"❌ Could not find an open port between {base_port} and {base_port + max_retries - 1}.")