Update app.py

app.py

@@ -72,10 +72,10 @@ def load_sample_problems():
             "What is the probability of rolling a 6 on a die 3 times in a row?"
         ]
         return math_samples
-        
+
 def create_math_system_message():
     """Specialized system prompt for mathematics with LaTeX"""
-    return """You are Mathetics AI, an advanced mathematics tutor and problem solver.
+    return r"""You are Mathetics AI, an advanced mathematics tutor and problem solver.
 
 🧮 **Your Expertise:**
 - Step-by-step problem solving with clear explanations
@@ -115,7 +115,7 @@ def render_latex(text):
     try:
         # Fix common LaTeX patterns from Qwen
         text = re.sub(r'(?<!\\)\$([^\$]+)\$(?!\$)', r'$\1$', text)
-        text = re.sub(r'\$\$([^\$]+)\$\$', r'$$\1$$', text)
+        text = re.sub(r'\$\$([^\$]+)\$$', r'$$\1$$', text)
         text = re.sub(r'\\\[([^\\]+)\\\]', r'$$\1$$', text)
         text = re.sub(r'\\\(([^\\]+)\\\)', r'$\1$', text)
         
@@ -135,206 +135,95 @@ def render_latex(text):
     return text
 
 def respond(message, history, system_message, max_tokens, temperature, top_p):
-    """Enhanced response with proper LaTeX streaming"""
-    yield "🤔 Thinking step-by-step..."
-    
+    """Non-streaming response for stability"""
     client = InferenceClient(model="Qwen/Qwen2.5-Math-7B-Instruct")
     
-    messages = []
-    if system_message:
-        messages.append({"role": "system", "content": system_message})
-    
-    # Convert history to proper format
-    for user_msg, assistant_msg in history:
-        if user_msg:
-            messages.append({"role": "user", "content": user_msg})
-        if assistant_msg:
-            messages.append({"role": "assistant", "content": assistant_msg})
-    
+    messages = [{"role": "system", "content": system_message}]
+    # Iterate over history dicts and add user/assistant pairs
+    for msg in history:
+        if msg["role"] == "user":
+            messages.append({"role": "user", "content": msg["content"]})
+        elif msg["role"] == "assistant":
+            messages.append({"role": "assistant", "content": msg["content"]})
     messages.append({"role": "user", "content": message})
     
-    response = ""
-    max_tokens = max(max_tokens, 1536)
-    
     try:
-        for message_chunk in client.chat_completion(
+        completion = client.chat_completion(
             messages,
             max_tokens=max_tokens,
-            stream=True,
             temperature=temperature,
             top_p=top_p,
-            timeout=60
-        ):
-            choices = message_chunk.choices
-            if len(choices) and choices[0].delta.content:
-                token = choices[0].delta.content
-                response += token
-                
-                if len(response) % 50 == 0 or token.strip() in ['.', '!', '?', '\n']:
-                    formatted = render_latex(response)
-                    yield formatted
-                else:
-                    yield response
-                    
-        final_formatted = render_latex(response.strip())
-        yield final_formatted
-            
+            timeout=30
+        )
+        response = completion.choices[0].message.content
+        return render_latex(response)
     except Exception as e:
-        error_msg = f"❌ **Error**: {str(e)[:100]}...\n\n💡 Try a simpler problem or wait a moment."
-        yield error_msg
+        return f"❌ Error: {str(e)[:100]}... Try a simpler problem."
 
 def get_random_sample():
     """Get a random sample problem - loads datasets if needed"""
     global math_samples
     if math_samples is None:
-        math_samples = load_sample_problems()  # ✅ FIXED: Now calls the right function
-    if math_samples:
-        return random.choice(math_samples)
-    return "Solve for x: 2x² + 5x - 3 = 0"
-    
+        math_samples = load_sample_problems()
+    return random.choice(math_samples)
+
 def insert_sample_to_chat(difficulty):
     """Insert random sample into chat input"""
-    sample = get_random_sample()
-    return sample
+    return get_random_sample()
 
 def show_help():
     return """**🧮 Math Help Tips:**
 
-1. **Be Specific**: "Find the derivative of x² + 3x" instead of "help with calculus"
-2. **Request Steps**: "Show me step-by-step how to solve..."
-3. **Ask for Verification**: "Check if my answer x=5 is correct"
-4. **Alternative Methods**: "What's another way to solve this integral?"
-5. **Use Clear Notation**: "lim(x->0)" for limits
-
-**Pro Tip**: Crank tokens to 1500+ for competition problems!"""
+1. Be Specific: "Find the derivative of x² + 3x" instead of "help with calculus"
+2. Request Steps: "Show me step-by-step how to solve..."
+3. Ask for Verification: "Check if my answer x=5 is correct"
+4. Alternative Methods: "What's another way to solve this integral?"
+5. Use Clear Notation: "lim(x->0)" for limits
 
-# CLASSIC Chatbot interface (Gradio 3.x compatible)
-def chat_response(message, history):
-    """Main chat function - compatible with all Gradio versions"""
-    bot_response = ""
-    for response in respond(
-        message, 
-        history, 
-        create_math_system_message(), 
-        1024,  # Default tokens
-        0.3,    # Default temperature
-        0.85    # Default top_p
-    ):
-        bot_response = response
-        history.append([message, bot_response])
-        yield history, ""
-    
-    return history, ""
+Pro Tip: Crank tokens to 1500+ for competition problems!"""
 
-# Build the interface
-with gr.Blocks(
-    title="🧮 Mathetics AI", 
-    theme=gr.themes.Soft(),
-    css="""
-    /* Enhanced math rendering */
-    .markdown-body { font-family: 'Times New Roman', Georgia, serif; line-height: 1.6; }
-    .katex { font-size: 1.1em !important; color: #2c3e50; }
-    .katex-display { font-size: 1.3em !important; text-align: center; margin: 1em 0; padding: 10px; background: #f8f9fa; border-radius: 8px; }
-    
-    /* Chat styling */
-    .message { margin: 10px 0; padding: 12px; border-radius: 8px; }
-    .user { background: linear-gradient(135deg, #e3f2fd 0%, #bbdefb 100%); border-left: 4px solid #2196f3; }
-    .assistant { background: linear-gradient(135deg, #f5f5f5 0%, #eeeeee 100%); border-left: 4px solid #4caf50; }
-    
-    /* Sidebar */
-    .difficulty-selector { background: linear-gradient(135deg, #fff3e0 0%, #ffe0b2 100%); padding: 15px; border-radius: 10px; margin: 10px 0; border: 1px solid #ffcc80; }
+# Simple Chatbot interface
+with gr.Blocks(title="🧮 Mathetics AI") as demo:
+    gr.Markdown("# 🧮 **Mathetics AI** - Math Tutor\nPowered by Qwen 2.5-Math")
     
-    /* Responsive */
-    @media (max-width: 768px) { .katex { font-size: 1em !important; } .katex-display { font-size: 1.1em !important; } }
-    """
-) as demo:
+    chatbot = gr.Chatbot(height=500, label="Conversation", type='messages')
+    help_text = gr.Markdown(visible=False)
     
-    gr.Markdown("""
-    # 🧮 **Mathetics AI** - Advanced Mathematics Solver
+    msg = gr.Textbox(placeholder="Ask a math problem...", show_label=False)
     
-    **Your Personal AI Math Tutor** | Step-by-step solutions with beautiful LaTeX rendering
-    
-    ---
-    """)
-    
-    # Main chat interface
-    chatbot = gr.Chatbot(
-        height=500,
-        show_label=False,
-        avatar_images=("🧑‍🎓", "🤖"),
-        bubble_full_width=False
-    )
-    
-    with gr.Row():
-        msg = gr.Textbox(
-            placeholder="Ask: 'Find the derivative of 3x² + 2x - 1'",
-            scale=4,
-            show_label=False,
-            lines=2
-        )
-        submit_btn = gr.Button("🚀 Solve", variant="primary", scale=1)
-    
-    # Controls
     with gr.Row():
-        with gr.Column(scale=2):
-            token_slider = gr.Slider(256, 2048, value=1024, step=128, label="📝 Max Tokens")
-            temp_slider = gr.Slider(0.1, 1.0, value=0.3, step=0.1, label="🎯 Temperature")
-            
-        with gr.Column(scale=1):
-            difficulty_preset = gr.Dropdown(
-                choices=["Elementary", "High School", "College", "Competition"],
-                value="High School",
-                label="🎯 Difficulty",
-                elem_classes=["difficulty-selector"]
-            )
-            sample_btn = gr.Button("🎲 Random Problem", variant="secondary")
-            help_btn = gr.Button("❓ Help", variant="secondary")
+        submit = gr.Button("Solve", variant="primary")
+        clear = gr.Button("Clear", variant="secondary")
+        sample = gr.Button("Random Problem", variant="secondary")
+        help_btn = gr.Button("Help", variant="secondary")
     
-    # Examples
     gr.Examples(
         examples=[
-            ["Find the derivative of f(x) = 3x² + 2x - 1"],
-            ["A triangle has sides 5, 12, and 13. What is its area?"],
-            ["Solve: lim(x->0) sin(x)/x"],
-            ["What is ∫(2x³ - 5x + 3) dx?"],
-            ["Solve the system: x + 2y = 7, 3x - y = 4"]
+            ["derivative of x^2 sin(x)"],
+            ["area of triangle 5-12-13"],
+            ["∫x^2 dx"]
         ],
-        inputs=msg,
-        label="💡 Quick Examples"
+        inputs=msg
     )
     
-    # Event handlers
-    def submit_message(message, history):
-        return chat_response(message, history)
+    def chat_response(message, history):
+        """Updated to use dict-based history for type='messages'."""
+        bot_response = respond(message, history, create_math_system_message(), 1024, 0.3, 0.85)
+        # Append as dicts, not tuples
+        history.append({"role": "user", "content": message})
+        history.append({"role": "assistant", "content": bot_response})
+        return history, ""
     
     def clear_chat():
+        """Clear the chat history and textbox."""
         return [], ""
     
-    msg.submit(submit_message, [msg, chatbot], [msg, chatbot])
-    submit_btn.click(submit_message, [msg, chatbot], [msg, chatbot])
-    
-    sample_btn.click(
-        insert_sample_to_chat,
-        inputs=[difficulty_preset],
-        outputs=msg
-    )
-    
-    help_btn.click(
-        show_help,
-        outputs=gr.Markdown(visible=True, label="Help")
+    msg.submit(chat_response, [msg, chatbot], [chatbot, msg])
+    submit.click(chat_response, [msg, chatbot], [chatbot, msg])
+    clear.click(clear_chat, outputs=[chatbot, msg])
+    sample.click(insert_sample_to_chat, outputs=msg)
+    help_btn.click(lambda: (show_help(), gr.update(visible=True)), outputs=[help_text, help_text]).then(
+        lambda: gr.update(visible=False), outputs=help_text
     )
-    
-    # Clear button
-    gr.Button("🗑️ Clear Chat", variant="secondary").click(
-        clear_chat,
-        outputs=[chatbot, msg]
-    )
-    
-    gr.Markdown("""
-    ---
-    **🔧 Tech:** Qwen2.5-Math-7B • LaTeX rendering • Streaming responses
-    **💡 Tip:** Use "lim(x->0)" for limits, crank tokens for complex problems
-    """)
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()