Update app.py

app.py

@@ -75,12 +75,12 @@ def process_input(input_type: str, query: str) -> Tuple[str, str]:
     """Process user input and return AI response."""
     if not query.strip():
         return "", "Please enter a query"
-    
+
     if input_type == "brainstorm":
         response = assistant.brainstorm_project(query)
     else:
         response = assistant.get_code_suggestion(query)
-    
+
     if response["success"]:
         status = f"Success! Tokens used: {response.get('tokens_used', 'N/A')}"
         return response["content"], status
@@ -91,9 +91,8 @@ def handle_project_click(project_name: str) -> Tuple[str, str, str]:
     """Handle when a project option is clicked."""
     description = PROJECT_OPTIONS[project_name]
     response = assistant.brainstorm_project(description)
-    
+
     if response["success"]:
-        status = f"Success! Tokens used: {response.get('tokens_used', 'N/A')}"
         return "brainstorm", description, response["content"]
     else:
         return "brainstorm", description, f"Error: {response.get('error', 'Unknown error occurred')}"
@@ -107,7 +106,7 @@ with gr.Blocks(css=CUSTOM_CSS, title="DDS AI Project Assistant") as interface:
                 <p class="subtitle">Your Generative AI Project Development Companion</p>
             </div>
         """)
-        
+
         with gr.Row():
             # Left sidebar with project options
             with gr.Column(scale=1, elem_classes="sidebar"):
@@ -116,13 +115,12 @@ with gr.Blocks(css=CUSTOM_CSS, title="DDS AI Project Assistant") as interface:
                         🎯 Popular GenAI Projects
                     </h3>
                 """)
-                project_buttons = [
-                    gr.Button(
-                        name,
-                        elem_classes="project-button"
-                    ) for name in PROJECT_OPTIONS.keys()
-                ]
-            
+                # Keep (name, button) pairs so we can wire up clicks correctly below
+                project_buttons = []
+                for name in PROJECT_OPTIONS.keys():
+                    btn = gr.Button(name, elem_classes="project-button")
+                    project_buttons.append((name, btn))
+
             # Main content area
             with gr.Column(scale=3):
                 input_type = gr.Radio(
@@ -130,18 +128,18 @@ with gr.Blocks(css=CUSTOM_CSS, title="DDS AI Project Assistant") as interface:
                     label="🤔 What kind of help do you need?",
                     value="brainstorm"
                 )
-                
+
                 query = gr.Textbox(
                     label="💭 Enter your topic or code request",
                     placeholder="e.g., 'text-to-image generation' or 'implement stable diffusion'",
                     elem_classes="output-box"
                 )
-                
+
                 submit_btn = gr.Button(
                     "🚀 Get AI Assistance",
                     elem_classes="project-button"
                 )
-                
+
                 with gr.Column(elem_classes="output-box"):
                     output = gr.Textbox(
                         label="🤖 AI Response",
@@ -150,27 +148,29 @@ with gr.Blocks(css=CUSTOM_CSS, title="DDS AI Project Assistant") as interface:
                     status = gr.Textbox(
                         label="📊 Status"
                     )
-        
+
         gr.HTML("""
             <div style="text-align: center; margin-top: 20px; padding: 10px; border-top: 2px solid #264653;">
                 <p>Developed by DDS Team | Powered by OpenAI</p>
             </div>
         """)
-    
+
     # Handle main submit button
     submit_btn.click(
         fn=process_input,
         inputs=[input_type, query],
         outputs=[output, status]
     )
-    
-    # Handle project option buttons
-    for btn in project_buttons:
+
+    # Handle project option buttons.
+    # A Button has no value to pass as an input, so we capture the project
+    # name in a default argument (n=name) and feed it to the handler directly.
+    for name, btn in project_buttons:
         btn.click(
-            fn=handle_project_click,
-            inputs=[btn],
+            fn=lambda n=name: handle_project_click(n),
+            inputs=None,
             outputs=[input_type, query, output]
         )
 
 if __name__ == "__main__":
-    interface.launch() 
+    interface.launch()