Update config.py

config.py

@@ -43,6 +43,16 @@ logger.info("📡 Connected to MongoDB synchronously")
 SECRET_KEY = os.getenv("SECRET_KEY", "your-secret-key")
 ALGORITHM = "HS256"
 
+
+
+# Determine WebSocket URL based on environment
+if os.getenv('SPACES') == '1':
+    # Hugging Face Spaces production
+    WS_URL = "wss://rocketfarmstudios-txagent-api.hf.space/queue/join"
+else:
+    # Local development
+    WS_URL = "ws://localhost:8000/ws/notifications"
+
 def setup_app(app: FastAPI):
     @app.on_event("startup")
     async def startup_event():