1

app.py

@@ -16,38 +16,48 @@ from datetime import datetime
 class GradioArchitectureApp:
     def __init__(self):
         load_dotenv()
-        api_key = os.getenv("OPENAI_API_KEY")
-        if not api_key:
-            raise ValueError("Please set OPENAI_API_KEY in .env file")
+        # For Hugging Face Spaces deployment - get API key from environment
+        api_key = os.environ.get("OPENAI_API_KEY")
         
+        # Make API key optional for HF Spaces - users can provide their own
         self.api_key = api_key
         self.assistant = None
         self.current_user_id = None
         self.conversation_history = []
         self.state_manager = user_state_manager
     
-    def initialize_user_session(self, user_identifier: str = None):
+    def initialize_user_session(self, user_identifier: str = None, api_key: str = None):
         """Initialize user session with persistent state"""
         if not user_identifier:
             # Generate user ID from timestamp for demo
             user_identifier = hashlib.md5(str(datetime.now()).encode()).hexdigest()[:12]
         
+        # Use provided API key or fallback to environment
+        effective_api_key = api_key or self.api_key
+        if not effective_api_key:
+            raise ValueError("No API key provided")
+            
         self.current_user_id = user_identifier
-        self.assistant = ArchitectureAssistant(self.api_key, user_identifier)
+        self.assistant = ArchitectureAssistant(effective_api_key, user_identifier)
         
         return user_identifier
     
-    def change_user_id(self, new_user_id: str):
+    def change_user_id(self, new_user_id: str, api_key: str = None):
         """Change user ID and auto-load their conversation history"""
         if not new_user_id.strip():
             return [], "❌ Please enter a valid User ID", "", "Start chatting to see your requirements here!"
         
         try:
+            # Use provided API key or fallback to environment
+            effective_api_key = api_key or self.api_key
+            if not effective_api_key:
+                return [], "❌ Please provide your OpenAI API key", "", "API key required"
+            
             # Set new user ID
             self.current_user_id = new_user_id.strip()
             
             # Create new assistant instance for this user
-            self.assistant = ArchitectureAssistant(self.api_key, self.current_user_id)
+            self.assistant = ArchitectureAssistant(effective_api_key, self.current_user_id)
             
             # Try to load their latest conversation
             history = self.state_manager.get_user_history(self.current_user_id)
@@ -89,18 +99,25 @@ class GradioArchitectureApp:
         except Exception as e:
             return [], f"❌ Error loading user {new_user_id}: {str(e)}", "", "Error loading user"
     
-    def chat_with_assistant(self, message, history, user_id_input=""):
+    def chat_with_assistant(self, message, history, user_id_input="", api_key=""):
         """Process user message and return response with state saving"""
         if not message.strip():
             return history, "", user_id_input, self.get_conversation_summary()
         
+        # Check for API key first
+        effective_api_key = api_key or self.api_key
+        if not effective_api_key:
+            error_msg = "❌ Please provide your OpenAI API key to use the Architecture Assistant."
+            history.append([message, error_msg])
+            return history, "", user_id_input, "API key required"
+        
         # Ensure we have an assistant instance
         if not self.assistant:
             if user_id_input.strip():
                 self.current_user_id = user_id_input.strip()
-                self.assistant = ArchitectureAssistant(self.api_key, self.current_user_id)
+                self.assistant = ArchitectureAssistant(effective_api_key, self.current_user_id)
             else:
-                user_id = self.initialize_user_session()
+                user_id = self.initialize_user_session(api_key=effective_api_key)
                 return history, "", user_id, self.get_conversation_summary()
         
         # Handle special commands
@@ -308,7 +325,15 @@ def create_gradio_interface():
         </div>
         """)
         
-        # User ID input at the top
+        # API Key input for Hugging Face Spaces
+        api_key_input = gr.Textbox(
+            label="🔑 OpenAI API Key",
+            placeholder="Enter your OpenAI API key (sk-...)",
+            type="password",
+            info="Required for AI functionality. Not stored or logged."
+        )
+        
+        # User ID input
         with gr.Row():
             with gr.Column(scale=4):
                 user_id_input = gr.Textbox(
@@ -421,11 +446,11 @@ def create_gradio_interface():
                     )
         
         # Event handlers
-        def send_message(message, history, user_id):
-            return app.chat_with_assistant(message, history, user_id)
+        def send_message(message, history, user_id, api_key):
+            return app.chat_with_assistant(message, history, user_id, api_key)
         
-        def change_user(user_id):
-            return app.change_user_id(user_id)
+        def change_user(user_id, api_key):
+            return app.change_user_id(user_id, api_key)
         
         def view_user_history(user_id):
             return app.get_user_history_display(user_id)
@@ -434,10 +459,10 @@ def create_gradio_interface():
             return app.get_all_users_summary()
         
         # Wire up the events
-        load_user_btn.click(change_user, [user_id_input], [chatbot, status_display, user_id_input, summary_display])
+        load_user_btn.click(change_user, [user_id_input, api_key_input], [chatbot, status_display, user_id_input, summary_display])
         
-        msg.submit(send_message, [msg, chatbot, user_id_input], [chatbot, msg, user_id_input, summary_display])
-        send_btn.click(send_message, [msg, chatbot, user_id_input], [chatbot, msg, user_id_input, summary_display])
+        msg.submit(send_message, [msg, chatbot, user_id_input, api_key_input], [chatbot, msg, user_id_input, summary_display])
+        send_btn.click(send_message, [msg, chatbot, user_id_input, api_key_input], [chatbot, msg, user_id_input, summary_display])
         
         clear_btn.click(lambda: app.assistant.reset_conversation() if app.assistant else None, outputs=[chatbot, summary_display])
         
@@ -457,25 +482,17 @@ def create_gradio_interface():
 
 
 def main():
-    """Launch the final enhanced Gradio interface"""
+    """Launch the enhanced Gradio interface for Hugging Face Spaces"""
     try:
         interface = create_gradio_interface()
-        print("🏠 Starting Final Architecture Assistant Web UI...")
+        print("🏠 Starting Architecture Assistant for Hugging Face Spaces...")
         print("👥 Features: Multi-user support, auto-loading conversations, precise floorplans")
-        print("📝 Make sure you have set OPENAI_API_KEY in your .env file")
-        print("🔧 Fixed issues: User conversation resumption, exact floorplan dimensions, SVG generation")
-        
-        interface.launch(
-            server_name="0.0.0.0",
-            server_port=7861,
-            share=False,
-            show_error=True,
-            quiet=False
-        )
+        print("🔑 Users provide their own OpenAI API key via the interface")
+        print("🚀 Ready for Hugging Face Spaces deployment!")
+        
+        # Simple launch for HF Spaces
+        interface.launch()
         
-    except ValueError as e:
-        print(f"❌ Configuration Error: {e}")
-        print("Please copy .env.example to .env and add your OpenAI API key")
     except Exception as e:
         print(f"❌ Error starting the application: {e}")