Add test/deploy env args

app.py

@@ -10,7 +10,7 @@ from loguru import logger
 from Gradio_UI import GradioUI
 from smolagents import LiteLLMModel
 
-from config import setup_logger, load_api_keys, get_model_id, GRADIO_CONFIG
+from config import setup_logger, load_api_keys, get_model_id, get_gradio_config
 from agents.manager_agent import create_manager_agent
 
 
@@ -35,12 +35,27 @@ def main():
     
     # Start the Gradio UI server
     logger.info("Initializing Gradio UI and launching server")
-    GradioUI(manager_agent).launch(
-        debug=GRADIO_CONFIG['debug'], 
-        share=GRADIO_CONFIG['share'],
-        server_name=GRADIO_CONFIG['server_name'], 
-        server_port=GRADIO_CONFIG['server_port']
-    )
+    
+    # Determine if we're in test mode (local) or production (HuggingFace)
+    # HuggingFace environment has SPACE_ID environment variable
+    import os
+    is_test = os.environ.get('SPACE_ID') is None
+    gradio_config = get_gradio_config(is_test)
+    
+    # Launch with appropriate configuration
+    launch_kwargs = {
+        "debug": gradio_config["debug"],
+        "share": gradio_config["share"]
+    }
+    
+    # Add server parameters only for local testing
+    if is_test:
+        launch_kwargs.update({
+            "server_name": gradio_config["server_name"],
+            "server_port": gradio_config["server_port"]
+        })
+    
+    GradioUI(manager_agent).launch(**launch_kwargs)
     logger.success("Server started successfully")
 
 

config.py

@@ -78,9 +78,28 @@ AGENT_CONFIG = {
 }
 
 # Gradio UI configuration
-GRADIO_CONFIG = {
-    "debug": True,
-    "share": True,
-    "server_name": "127.0.0.1",
-    "server_port": 3000
-}
+def get_gradio_config(is_test=True):
+    """Get the appropriate Gradio UI configuration based on environment.
+    
+    Args:
+        is_test: If True, use test configuration (local development).
+               If False, use production configuration (HuggingFace).
+    
+    Returns:
+        Dictionary with Gradio configuration parameters.
+    """
+    if is_test:
+        # Configuration for local development/testing
+        return {
+            "debug": True,
+            "share": False,
+            "server_name": "127.0.0.1",
+            "server_port": 3000
+        }
+    else:
+        # Configuration for production (HuggingFace)
+        return {
+            "debug": True,
+            "share": False
+            # No server_name or server_port for HuggingFace deployment
+        }