Add error handling back to create_demo_for_spaces

app.py

@@ -834,47 +834,57 @@ def main():
 # Spaces will import this module and look for a 'demo' variable
 def create_demo_for_spaces():
     """Create demo for Hugging Face Spaces - called at module level"""
-    # Initialize with default args for Spaces
-    parser = argparse.ArgumentParser()
-    parser.add_argument('--model', type=str, default='meta-llama/Llama-3.2-3B-Instruct')
-    parser.add_argument('--vector-db-dir', default='./chroma_db')
-    parser.add_argument('--data-dir', default='./Data Resources')
-    parser.add_argument('--max-new-tokens', type=int, default=1024)
-    parser.add_argument('--temperature', type=float, default=0.2)
-    parser.add_argument('--top-p', type=float, default=0.9)
-    parser.add_argument('--repetition-penalty', type=float, default=1.1)
-    parser.add_argument('--k', type=int, default=5)
-    parser.add_argument('--skip-indexing', action='store_true', default=True)
-    parser.add_argument('--verbose', action='store_true', default=False)
-    parser.add_argument('--share', action='store_true', default=False)
-    parser.add_argument('--server-name', type=str, default='0.0.0.0')
-    parser.add_argument('--server-port', type=int, default=7860)
-    parser.add_argument('--seed', type=int, default=42)
-    
-    args = parser.parse_args([])  # Empty args for Spaces
-    args.skip_model_loading = True  # Skip model loading, use Inference API
-    
-    # Create bot (will skip model loading)
-    bot = RAGBot(args)
-    
-    if bot.vector_retriever is None:
-        raise Exception("Vector database not available")
-    
-    # Create and return the demo interface
-    return create_interface(bot, use_inference_api=True)
+    try:
+        # Initialize with default args for Spaces
+        parser = argparse.ArgumentParser()
+        parser.add_argument('--model', type=str, default='meta-llama/Llama-3.2-3B-Instruct')
+        parser.add_argument('--vector-db-dir', default='./chroma_db')
+        parser.add_argument('--data-dir', default='./Data Resources')
+        parser.add_argument('--max-new-tokens', type=int, default=1024)
+        parser.add_argument('--temperature', type=float, default=0.2)
+        parser.add_argument('--top-p', type=float, default=0.9)
+        parser.add_argument('--repetition-penalty', type=float, default=1.1)
+        parser.add_argument('--k', type=int, default=5)
+        parser.add_argument('--skip-indexing', action='store_true', default=True)
+        parser.add_argument('--verbose', action='store_true', default=False)
+        parser.add_argument('--share', action='store_true', default=False)
+        parser.add_argument('--server-name', type=str, default='0.0.0.0')
+        parser.add_argument('--server-port', type=int, default=7860)
+        parser.add_argument('--seed', type=int, default=42)
+        
+        args = parser.parse_args([])  # Empty args for Spaces
+        args.skip_model_loading = True  # Skip model loading, use Inference API
+        
+        # Create bot (will skip model loading)
+        bot = RAGBot(args)
+        
+        if bot.vector_retriever is None:
+            raise Exception("Vector database not available")
+        
+        # Create and return the demo interface
+        demo_obj = create_interface(bot, use_inference_api=True)
+        logger.info("Demo interface created successfully")
+        return demo_obj
+    except Exception as e:
+        logger.error(f"Error in create_demo_for_spaces: {e}", exc_info=True)
+        # Return a simple error demo so Spaces doesn't crash
+        with gr.Blocks() as error_demo:
+            gr.Markdown(f"# Error\n\n{str(e)}")
+        return error_demo
 
 # For Hugging Face Spaces: create demo at module level
 # Spaces will import this module and look for a 'demo' variable
 # Always create demo - Spaces will use it, local execution will override in main()
 
-# Check if we're on Spaces
-IS_SPACES = os.getenv("SPACE_ID") is not None or os.getenv("SYSTEM") == "spaces"
-
 # For Hugging Face Spaces: create demo at module level
 # Spaces will import this module and look for a 'demo' variable
 # Pattern: demo = gr.Interface(...) or demo = gr.Blocks(...)
 # DO NOT call demo.launch() - Spaces handles that automatically
 
+# Check if we're on Spaces
+IS_SPACES = os.getenv("SPACE_ID") is not None or os.getenv("SYSTEM") == "spaces"
+
+# Create demo - Spaces will use it, local execution will override in main()
 if IS_SPACES:
     logger.info("Initializing for Hugging Face Spaces...")
     demo = create_demo_for_spaces()