Update app.py

app.py

@@ -120,12 +120,13 @@ def format_response_with_thinking(response):
 
 def chat_submit(message, history, conversation_state, model_name, max_length, temperature):
     """Process a new message and update the chat history"""
+    # For debugging - print when the function is called
+    print(f"chat_submit function called with message: '{message}'")
+    
     if not message or not message.strip():
+        print("Empty message, returning without processing")
         return "", history, conversation_state
     
-    # Debug print to check function execution
-    print(f"Processing message in chat_submit: {message}")
-    
     model_id = MODELS.get(model_name, MODELS["Athena-R3X 4B"])
     try:
         response, load_time, generation_time = generate_response(
@@ -224,9 +225,8 @@ css = """
 }
 """
 
-# Add JavaScript to make the thinking buttons work and fix enter key issues
+# Add JavaScript to make the thinking buttons work
 js = """
-// Function to handle thinking toggle buttons
 function setupThinkingToggle() {
     document.querySelectorAll('.thinking-toggle').forEach(button => {
         if (!button.hasEventListener) {
@@ -252,45 +252,9 @@ const observer = new MutationObserver(function(mutations) {
     setupThinkingToggle();
 });
 
-// Function to ensure the textbox and submit button work correctly
-function fixChatInputs() {
-    const textbox = document.querySelector('textarea[data-testid="textbox"]');
-    const submitBtn = document.querySelector('button[data-testid="send-btn"]');
-    
-    if (textbox && !textbox.hasEnterListener) {
-        console.log("Setting up enter key handler");
-        textbox.addEventListener('keydown', function(e) {
-            if (e.key === 'Enter' && !e.shiftKey) {
-                e.preventDefault();
-                if (textbox.value.trim() !== '') {
-                    submitBtn.click();
-                }
-            }
-        });
-        textbox.hasEnterListener = true;
-    }
-    
-    if (submitBtn && !submitBtn.hasClickFix) {
-        console.log("Enhancing submit button");
-        submitBtn.addEventListener('click', function() {
-            console.log("Submit button clicked");
-        });
-        submitBtn.hasClickFix = true;
-    }
-}
-
-// Function to run all UI fixes
-function setupUI() {
-    setupThinkingToggle();
-    fixChatInputs();
-}
-
-// Initial setup
+// Start observing after DOM is loaded
 document.addEventListener('DOMContentLoaded', () => {
-    console.log("DOM loaded, setting up UI");
-    setTimeout(setupUI, 1000);
-    
-    // Set up observer after a delay
+    setupThinkingToggle();
     setTimeout(() => {
         const chatbot = document.querySelector('.chatbot');
         if (chatbot) {
@@ -300,49 +264,47 @@ document.addEventListener('DOMContentLoaded', () => {
                 characterData: true
             });
         } else {
-            // If chatbot container not found, observe the body
             observer.observe(document.body, { 
                 childList: true, 
                 subtree: true 
             });
         }
-        
-        // Run UI fixes periodically
-        setInterval(setupUI, 2000);
     }, 1000);
 });
 """
 
-theme = gr.themes.Soft()
-
-with gr.Blocks(title="Athena Playground Chat", css=css, theme=theme, js=js) as demo:
+# Create Gradio interface
+with gr.Blocks(title="Athena Playground Chat", css=css, js=js) as demo:
     gr.Markdown("# 🚀 Athena Playground Chat")
     gr.Markdown("*Powered by HuggingFace ZeroGPU*")
 
     # State to keep track of the conversation for the model
     conversation_state = gr.State([])
     
-    chatbot = gr.Chatbot(height=500, label="Athena", render_markdown=True, elem_classes=["chatbot"])
+    # Chatbot component
+    chatbot = gr.Chatbot(
+        height=500, 
+        label="Athena", 
+        render_markdown=True, 
+        elem_classes=["chatbot"]
+    )
     
+    # Input and send button row
     with gr.Row():
         user_input = gr.Textbox(
             label="Your message", 
             scale=8, 
             autofocus=True,
             placeholder="Type your message here...",
-            elem_id="chat-input",
-            lines=2,
-            max_lines=10,
+            lines=2
         )
         send_btn = gr.Button(
             value="Send", 
             scale=1, 
-            variant="primary",
-            elem_id="send-btn",
-            min_width=100
+            variant="primary"
         )
 
-    # Clear button for resetting the conversation
+    # Clear button
     clear_btn = gr.Button("Clear Conversation")
 
     # Configuration controls
@@ -369,30 +331,27 @@ with gr.Blocks(title="Athena Playground Chat", css=css, theme=theme, js=js) as d
     def clear_conversation():
         return [], []
     
-    # Connect the interface components - note the specific ordering
-    user_input.submit(
+    # Connect the interface components with explicit handlers
+    submit_click = user_input.submit(
         fn=chat_submit,
         inputs=[user_input, chatbot, conversation_state, model_choice, max_length, temperature],
-        outputs=[user_input, chatbot, conversation_state],
-        api_name="submit_message"
+        outputs=[user_input, chatbot, conversation_state]
     )
     
-    # Make sure send button uses the exact same function with the same parameter ordering
-    send_btn.click(
+    # Connect send button explicitly
+    send_click = send_btn.click(
         fn=chat_submit,
         inputs=[user_input, chatbot, conversation_state, model_choice, max_length, temperature],
-        outputs=[user_input, chatbot, conversation_state],
-        api_name="send_message"
+        outputs=[user_input, chatbot, conversation_state]
     )
     
-    # Connect clear button
+    # Clear conversation
     clear_btn.click(
         fn=clear_conversation, 
-        outputs=[chatbot, conversation_state],
-        api_name="clear_chat"
+        outputs=[chatbot, conversation_state]
     )
 
-    # Add examples if desired
+    # Examples
     gr.Examples(
         examples=[
             "What is artificial intelligence?",
@@ -400,7 +359,7 @@ with gr.Blocks(title="Athena Playground Chat", css=css, theme=theme, js=js) as d
             "Write a short poem about technology",
             "What are some ethical concerns about AI?"
         ],
-        inputs=[user_input]
+        inputs=user_input
     )
 
     gr.Markdown("""
@@ -410,5 +369,6 @@ with gr.Blocks(title="Athena Playground Chat", css=css, theme=theme, js=js) as d
     """)
 
 if __name__ == "__main__":
-    demo.queue()  # Enable queuing for smoother experience
-    demo.launch(debug=True)  # Enable debug mode for better error reporting
+    # Enable queue and debugging
+    demo.queue()
+    demo.launch(debug=True)