Upload app.py

app.py

@@ -136,23 +136,59 @@ def create_ephemeral_session(credentials: HTTPBasicCredentials = Depends(securit
     return data
 
 # Streamlit app
+def find_available_port(start_port=5050, max_attempts=10):
+    """Find an available port starting from start_port"""
+    import socket
+    
+    for port in range(start_port, start_port + max_attempts):
+        try:
+            with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
+                s.bind(('0.0.0.0', port))
+                return port
+        except OSError:
+            continue
+    return None
+
 def start_token_server_once():
     if getattr(st.session_state, "_token_server_started", False):
         return
     st.session_state._token_server_started = True
 
     def _run():
-        os.environ.setdefault("TOKEN_SERVER_PORT", "5050")
-        uvicorn.run(
-            app,
-            host="0.0.0.0",
-            port=int(os.getenv("TOKEN_SERVER_PORT", "5050")),
-            log_level="warning",
-        )
+        # Find an available port
+        port = find_available_port(5050)
+        if port is None:
+            st.error("❌ No available ports found. Please close other applications using ports 5050-5059")
+            return
+        
+        # Update the port in session state so the iframe uses the correct port
+        st.session_state.token_server_port = port
+        
+        try:
+            uvicorn.run(
+                app,
+                host="0.0.0.0",
+                port=port,
+                log_level="warning",
+            )
+        except OSError as e:
+            if "address already in use" in str(e):
+                st.error(f"❌ Port {port} is already in use. Please close other applications or restart your browser.")
+            else:
+                st.error(f"❌ Server error: {e}")
 
     t = threading.Thread(target=_run, daemon=True)
     t.start()
 
+def check_server_health(port):
+    """Check if the FastAPI server is running and healthy"""
+    try:
+        import requests
+        response = requests.get(f"http://localhost:{port}/health", timeout=1)
+        return response.status_code == 200
+    except:
+        return False
+
 # Streamlit UI
 st.set_page_config(page_title="Catherine – Role-Play Voice Simulator", page_icon="🎤", layout="centered")
 st.title("Client Role-Play Voice Simulator (Catherine)")
@@ -166,9 +202,35 @@ missing_key = os.getenv("OPENAI_API_KEY") is None
 if missing_key:
     st.warning("Set environment variable OPENAI_API_KEY before starting.")
 
+# Start the server
 start_token_server_once()
 
-# Embed the static client via iframe
-token_server_port = int(os.getenv("TOKEN_SERVER_PORT", "5050"))
-client_url = f"http://localhost:{token_server_port}/static/index.html"
-st.components.v1.iframe(client_url, height=760)
+# Get the port (either from session state or environment)
+token_server_port = st.session_state.get("token_server_port", int(os.getenv("TOKEN_SERVER_PORT", "5050")))
+
+# Use a placeholder while server starts
+if "server_ready" not in st.session_state:
+    st.session_state.server_ready = False
+
+# Check if server is ready
+if not st.session_state.server_ready:
+    with st.spinner("Starting voice server..."):
+        import time
+        time.sleep(2)  # Give server time to start
+        st.session_state.server_ready = check_server_health(token_server_port)
+
+if st.session_state.server_ready:
+    # Server is ready, show the iframe
+    client_url = f"http://localhost:{token_server_port}/static/index.html"
+    st.components.v1.iframe(client_url, height=760)
+else:
+    # Server not ready, show retry option
+    st.warning("Voice server is starting up...")
+    if st.button("🔄 Retry Connection"):
+        st.session_state.server_ready = check_server_health(token_server_port)
+        st.rerun()
+    
+    st.info("💡 **Troubleshooting:** If this persists, try refreshing the page or running from command line: `streamlit run app.py`")
+    
+    # Show port information for debugging
+    st.code(f"Trying to connect to port: {token_server_port}")