Update app.py

app.py

@@ -34,14 +34,14 @@ custom_css = """
 }
 """
 
-# State variable to track current model
-current_model = gr.State("openai/gpt-oss-120b")
-
-def switch_model(model_choice):
-    """Function to switch between models"""
-    return gr.update(visible=False), gr.update(visible=True), model_choice
+def chat_fn(message, history, model_choice):
+    """Placeholder chat function"""
+    return f"Response from {model_choice}: {message}"
 
 with gr.Blocks(fill_height=True, theme="soft", css=custom_css) as demo:
+    # State variable to track current model
+    current_model = gr.State("openai/gpt-oss-120b")
+    
     with gr.Row():
         # Sidebar
         with gr.Column(scale=1):
@@ -66,9 +66,6 @@ with gr.Blocks(fill_height=True, theme="soft", css=custom_css) as demo:
                 # Login button
                 login_button = gr.LoginButton("Sign in with Hugging Face", size="lg")
                 
-                # Reload button to apply model change
-                reload_btn = gr.Button("🔄 Apply Model Change", variant="primary", size="lg")
-                
                 # Additional options
                 with gr.Accordion("⚙️ Advanced Options", open=False):
                     gr.Markdown("*These options will be available after model implementation*")
@@ -92,37 +89,77 @@ with gr.Blocks(fill_height=True, theme="soft", css=custom_css) as demo:
             with gr.Group(elem_classes="main-container"):
                 gr.Markdown("## 💬 Chat Interface")
                 
-                # Container for model interfaces
-                with gr.Column(visible=True) as model_120b_container:
-                    gr.Markdown("### Model: openai/gpt-oss-120b")
-                    gr.load("models/openai/gpt-oss-120b", accept_token=login_button, provider="fireworks-ai")
+                # Display current model
+                model_display = gr.Markdown(f"### Model: openai/gpt-oss-120b")
+                
+                # Single chat interface that works with both models
+                chatbot = gr.Chatbot(
+                    height=400,
+                    show_label=False,
+                    elem_classes="main-container"
+                )
+                
+                with gr.Row():
+                    msg = gr.Textbox(
+                        placeholder="Type your message here...",
+                        show_label=False,
+                        scale=9
+                    )
+                    submit_btn = gr.Button("Send", variant="primary", scale=1)
                 
-                with gr.Column(visible=False) as model_20b_container:
-                    gr.Markdown("### Model: openai/gpt-oss-20b")
-                    gr.load("models/openai/gpt-oss-20b", accept_token=login_button, provider="fireworks-ai")
+                clear_btn = gr.Button("🗑️ Clear Chat", variant="secondary")
     
-    # Handle model switching
-    reload_btn.click(
-        fn=switch_model,
+    # Update model display when dropdown changes
+    def update_model_display(model_choice):
+        return f"### Model: {model_choice}", model_choice
+    
+    model_dropdown.change(
+        fn=update_model_display,
         inputs=[model_dropdown],
-        outputs=[model_120b_container, model_20b_container, current_model]
-    ).then(
-        fn=lambda: gr.Info("Model switched successfully!"),
-        inputs=[],
-        outputs=[]
+        outputs=[model_display, current_model]
+    )
+    
+    # Chat functionality
+    def respond(message, chat_history, model):
+        if not message:
+            return "", chat_history
+        
+        # Add user message to history
+        chat_history = chat_history + [[message, None]]
+        
+        # Generate bot response (placeholder - replace with actual model call)
+        bot_response = f"[{model.split('/')[-1]}]: This is a placeholder response. In production, this would connect to the {model} via Fireworks AI API."
+        
+        # Update the last message with bot response
+        chat_history[-1][1] = bot_response
+        
+        return "", chat_history
+    
+    # Submit message on button click or enter
+    submit_btn.click(
+        fn=respond,
+        inputs=[msg, chatbot, current_model],
+        outputs=[msg, chatbot]
     )
     
-    # Update visibility based on dropdown selection
-    def update_visibility(model_choice):
-        if model_choice == "openai/gpt-oss-120b":
-            return gr.update(visible=True), gr.update(visible=False)
-        else:
-            return gr.update(visible=False), gr.update(visible=True)
+    msg.submit(
+        fn=respond,
+        inputs=[msg, chatbot, current_model],
+        outputs=[msg, chatbot]
+    )
     
-    model_dropdown.change(
-        fn=update_visibility,
-        inputs=[model_dropdown],
-        outputs=[model_120b_container, model_20b_container]
+    # Clear chat
+    clear_btn.click(
+        fn=lambda: (None, ""),
+        inputs=[],
+        outputs=[chatbot, msg]
+    )
+    
+    # Show info message on login
+    login_button.click(
+        fn=lambda: gr.Info("Please authenticate with Hugging Face to access the models"),
+        inputs=[],
+        outputs=[]
     )
 
 demo.launch()