Update app.py

app.py

@@ -105,13 +105,17 @@ def format_response_with_thinking(response):
     # If no thinking tags, return the original response
     return response
 
-def chat_submit(message, chat_history, conversation_state, model_name, max_length, temperature):
+def chat_submit(message, history, conversation_state, model_name, max_length, temperature):
     """Process a new message and update the chat history"""
     if not message.strip():
-        return "", chat_history, conversation_state
+        return "", history, conversation_state
     
     model_id = MODELS.get(model_name, MODELS["Athena-R3X 4B"])
     try:
+        # Print debug info to help diagnose issues
+        print(f"Processing message: {message}")
+        print(f"Selected model: {model_name} ({model_id})")
+        
         response, load_time, generation_time = generate_response(
             model_id, conversation_state, message, max_length, temperature
         )
@@ -124,13 +128,17 @@ def chat_submit(message, chat_history, conversation_state, model_name, max_lengt
         formatted_response = format_response_with_thinking(response)
         
         # Update the visible chat history
-        chat_history.append((message, formatted_response))
+        history.append((message, formatted_response))
+        print(f"Response added to history. Current length: {len(history)}")
         
-        return "", chat_history, conversation_state
+        return "", history, conversation_state
     except Exception as e:
+        import traceback
+        print(f"Error in chat_submit: {str(e)}")
+        print(traceback.format_exc())
         error_message = f"Error: {str(e)}"
-        chat_history.append((message, error_message))
-        return "", chat_history, conversation_state
+        history.append((message, error_message))
+        return "", history, conversation_state
 
 css = """
 .message {
@@ -171,27 +179,9 @@ css = """
 }
 """
 
-# Add JavaScript to handle the toggle functionality
-js = """
-function setupThinkingToggles() {
-    document.querySelectorAll('.thinking-toggle').forEach(button => {
-        button.addEventListener('click', function() {
-            const content = this.nextElementSibling;
-            content.classList.toggle('hidden');
-            this.textContent = content.classList.contains('hidden') ? 'Show reasoning' : 'Hide reasoning';
-        });
-    });
-}
-
-// Run after the page loads and when the chat updates
-document.addEventListener('DOMContentLoaded', setupThinkingToggles);
-const observer = new MutationObserver(setupThinkingToggles);
-observer.observe(document.body, { childList: true, subtree: true });
-"""
-
 theme = gr.themes.Monochrome()
 
-with gr.Blocks(title="Athena Playground Chat", css=css, theme=theme, js=js) as demo:
+with gr.Blocks(title="Athena Playground Chat", css=css, theme=theme) as demo:
     gr.Markdown("# 🚀 Athena Playground Chat")
     gr.Markdown("*Powered by HuggingFace ZeroGPU*")
 
@@ -204,6 +194,9 @@ with gr.Blocks(title="Athena Playground Chat", css=css, theme=theme, js=js) as d
         user_input = gr.Textbox(label="Your message", scale=8, autofocus=True, placeholder="Type your message here...")
         send_btn = gr.Button(value="Send", scale=1, variant="primary")
 
+    # Clear button for resetting the conversation
+    clear_btn = gr.Button("Clear Conversation")
+
     # Configuration controls
     gr.Markdown("### ⚙️ Model & Generation Settings")
     with gr.Row():
@@ -214,7 +207,7 @@ with gr.Blocks(title="Athena Playground Chat", css=css, theme=theme, js=js) as d
             info="Select which Athena model to use"
         )
         max_length = gr.Slider(
-            32, 8000, value=512, 
+            32, 2048, value=512, 
             label="📝 Max Tokens",
             info="Maximum number of tokens to generate"
         )
@@ -224,18 +217,26 @@ with gr.Blocks(title="Athena Playground Chat", css=css, theme=theme, js=js) as d
             info="Higher values = more creative responses"
         )
 
-    # Connect the interface components
-    submit_event = user_input.submit(
+    # Function to clear the conversation
+    def clear_conversation():
+        return [], []
+    
+    # Connect the interface components - note the specific ordering
+    user_input.submit(
         chat_submit,
         inputs=[user_input, chatbot, conversation_state, model_choice, max_length, temperature],
         outputs=[user_input, chatbot, conversation_state]
     )
     
+    # Make sure send button uses the exact same function with the same parameter ordering
     send_btn.click(
         chat_submit,
         inputs=[user_input, chatbot, conversation_state, model_choice, max_length, temperature],
         outputs=[user_input, chatbot, conversation_state]
     )
+    
+    # Connect clear button
+    clear_btn.click(clear_conversation, outputs=[chatbot, conversation_state])
 
     # Add examples if desired
     gr.Examples(
@@ -255,4 +256,4 @@ with gr.Blocks(title="Athena Playground Chat", css=css, theme=theme, js=js) as d
     """)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(debug=True)  # Enable debug mode for better error reporting