Update app.py

app.py

@@ -6,7 +6,7 @@ import random
 import re
 
 # Load math datasets for sample problems
-print("Loading datasets...")
+print("🔄 Loading datasets...")
 fw = load_dataset("HuggingFaceFW/fineweb-edu", name="sample-10BT", split="train", streaming=True)
 ds = load_dataset("HuggingFaceH4/ultrachat_200k", streaming=True)
 
@@ -59,12 +59,11 @@ def create_math_system_message():
 
 **LaTeX Guidelines:**
 - Use $...$ for inline math: $x^2 + y^2 = z^2$
-- Use $$...$$ or \[...\] for display math
+- Use $$...$$ for display math
 - Box final answers: \boxed{answer}
 - Fractions: \frac{numerator}{denominator}
 - Limits: \lim_{x \to 0}
 - Derivatives: \frac{d}{dx} or f'(x)
-- Integrals: \int f(x) \, dx
 
 Always be precise, educational, and encourage mathematical thinking."""
 
@@ -75,21 +74,16 @@ def render_latex(text):
     
     try:
         # Fix common LaTeX patterns from Qwen
-        # Ensure proper inline math delimiters
         text = re.sub(r'(?<!\\)\$([^\$]+)\$(?!\$)', r'$\1$', text)
-        
-        # Convert display math
         text = re.sub(r'\$\$([^\$]+)\$\$', r'$$\1$$', text)
         text = re.sub(r'\\\[([^\\]+)\\\]', r'$$\1$$', text)
         text = re.sub(r'\\\(([^\\]+)\\\)', r'$\1$', text)
         
         # Fix escaped LaTeX commands
-        text = re.sub(r'\\(lim|frac|sqrt|int|sum|prod|partial|nabla|infty|to|le|ge|neq|approx|cdot|times|div|deg|prime|log|ln|log|sin|cos|tan|cot|sec|csc|arcsin|arccos|arctan|sinh|cosh)', r'\1', text)
+        text = re.sub(r'\\(lim|frac|sqrt|int|sum|prod|partial|nabla|infty|to|le|ge|neq|approx|cdot|times|div|deg|prime|log|ln|sin|cos|tan|cot|sec|csc|arcsin|arccos|arctan|sinh|cosh)', r'\1', text)
         
         # Ensure boxed answers render
         text = re.sub(r'\\boxed\{([^}]+)\}', r'$$\boxed{\1}$$', text)
-        
-        # Fix fractions
         text = re.sub(r'\\frac\{([^}]+)\}\{([^}]+)\}', r'$\frac{\1}{\2}$', text)
         
         # Clean up extra spaces
@@ -97,34 +91,30 @@ def render_latex(text):
         
     except Exception as e:
         print(f"⚠️ LaTeX formatting error: {e}")
-        # Return original if processing fails
     
     return text
 
 def respond(message, history, system_message, max_tokens, temperature, top_p):
     """Enhanced response with proper LaTeX streaming"""
-    # Show initial thinking
     yield "🤔 Thinking step-by-step..."
     
-    # Use Qwen Math model
     client = InferenceClient(model="Qwen/Qwen2.5-Math-7B-Instruct")
     
-    # Build messages properly
     messages = []
     if system_message:
         messages.append({"role": "system", "content": system_message})
     
-    # Add conversation history
-    for msg in history:
-        if msg.get("role") == "user":
-            messages.append({"role": "user", "content": msg["content"]})
-        elif msg.get("role") == "assistant":
-            messages.append({"role": "assistant", "content": msg["content"]})
+    # 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.append({"role": "user", "content": message})
     
     response = ""
-    max_tokens = max(max_tokens, 1536)  # Ensure no truncation
+    max_tokens = max(max_tokens, 1536)
     
     try:
         for message_chunk in client.chat_completion(
@@ -133,26 +123,24 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
             stream=True,
             temperature=temperature,
             top_p=top_p,
-            timeout=60  # Add timeout for stability
+            timeout=60
         ):
             choices = message_chunk.choices
             if len(choices) and choices[0].delta.content:
                 token = choices[0].delta.content
                 response += token
                 
-                # Stream with formatting every few tokens
                 if len(response) % 50 == 0 or token.strip() in ['.', '!', '?', '\n']:
                     formatted = render_latex(response)
                     yield formatted
                 else:
                     yield response
                     
-        # Final formatted response
         final_formatted = render_latex(response.strip())
         yield final_formatted
             
     except Exception as e:
-        error_msg = f"❌ **Error**: {str(e)[:100]}...\n\n💡 **Troubleshooting**:\n• Try a simpler problem\n• Check your connection\n• Wait a moment and retry"
+        error_msg = f"❌ **Error**: {str(e)[:100]}...\n\n💡 Try a simpler problem or wait a moment."
         yield error_msg
 
 def get_random_sample():
@@ -164,7 +152,7 @@ def get_random_sample():
 def insert_sample_to_chat(difficulty):
     """Insert random sample into chat input"""
     sample = get_random_sample()
-    return sample, ""  # Return sample for input, clear status
+    return sample
 
 def show_help():
     return """**🧮 Math Help Tips:**
@@ -173,181 +161,137 @@ def show_help():
 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)" instead of arrows for mobile
-
-**LaTeX Examples:**
-- Inline: $x^2 + y^2 = z^2$
-- Display: $$\int x^2 \, dx = \frac{x^3}{3} + C$$
-- Boxed: $$\boxed{x = 5}$$
+5. **Use Clear Notation**: "lim(x->0)" for limits
 
 **Pro Tip**: Crank tokens to 1500+ for competition problems!"""
 
-# Enhanced ChatInterface with better LaTeX support
-chatbot = gr.ChatInterface(
-    respond,
-    type="messages",
-    title="🧮 **Mathetics AI** - Advanced Mathematics Solver",
-    description="""
-    **Powered by Qwen 2.5-Math** | **Specialized for Mathematical Problem Solving**
+# 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, ""
     
-    ✨ **Capabilities**: Algebra • Geometry • Calculus • Statistics • Competition Math
-    📚 **Features**: Step-by-step solutions • Multiple approaches • Beautiful LaTeX rendering
-    """,
-    additional_inputs=[
-        gr.Textbox(
-            value=create_math_system_message(), 
-            label="🧠 System Message (Math Tutor Personality)",
-            lines=4,
-            max_lines=12
-        ),
-        gr.Slider(minimum=256, maximum=2048, value=1024, step=128, label="📝 Max Tokens (Longer = More Detail)"),
-        gr.Slider(minimum=0.1, maximum=1.0, value=0.3, step=0.1, label="🎯 Temperature (Creativity vs Precision)"),
-        gr.Slider(minimum=0.1, maximum=1.0, value=0.85, step=0.05, label="🔍 Top-p (Response Diversity)"),
-    ],
-    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"]
-    ],
-    cache_examples=False,
-    concurrency_limit=5,  # Reduce for stability
-    retry_btn="🔄 Retry",
-    undo_btn="↶ Undo",
-    clear_btn="🗑️ Clear"
-)
+    return history, ""
 
-# Main interface with enhanced LaTeX CSS
+# 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 math rendering */
-    .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;
-    }
+    .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; }
     
-    /* Boxed answers */
-    .katex .mord.text { font-weight: bold; }
+    /* 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; }
     
-    /* Chat message 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; }
     
-    /* Sidebar styling */
-    .difficulty-selector { 
-        background: linear-gradient(135deg, #fff3e0 0%, #ffe0b2 100%);
-        padding: 15px; 
-        border-radius: 10px; 
-        margin: 10px 0;
-        border: 1px solid #ffcc80;
-    }
-    
-    .math-highlight { 
-        background: linear-gradient(135deg, #e8f5e8 0%, #c8e6c9 100%);
-        padding: 12px; 
-        border-left: 4px solid #4caf50; 
-        margin: 10px 0; 
-        border-radius: 8px;
-    }
-    
-    /* Responsive design */
-    @media (max-width: 768px) {
-        .katex { font-size: 1em !important; }
-        .katex-display { font-size: 1.1em !important; }
-    }
+    /* Responsive */
+    @media (max-width: 768px) { .katex { font-size: 1em !important; } .katex-display { font-size: 1.1em !important; } }
     """
 ) as demo:
     
     gr.Markdown("""
     # 🧮 **Mathetics AI** - Advanced Mathematics Solver
     
-    **Your Personal AI Math Tutor** | Specialized in step-by-step problem solving with beautiful LaTeX rendering
+    **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=4):
-            chatbot.render()
+        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):
-            with gr.Accordion("🎲 **Quick Actions**", open=True):
-                difficulty_preset = gr.Dropdown(
-                    choices=["Elementary", "High School", "College", "Competition"],
-                    value="High School",
-                    label="🎯 Problem Difficulty",
-                    elem_classes=["difficulty-selector"]
-                )
-                
-                sample_btn = gr.Button("🎯 Get Sample Problem", variant="secondary", size="sm")
-                help_btn = gr.Button("❓ Math Help Tips", variant="secondary", size="sm")
-                
-                gr.Markdown("### 🔧 **Quick Tools**")
-                gr.Markdown("""
-                - **Algebra**: Equations, inequalities, factoring
-                - **Geometry**: Area, volume, trigonometry  
-                - **Calculus**: Derivatives, integrals, limits
-                - **Statistics**: Probability, distributions
-                - **Number Theory**: Prime factorization, GCD/LCM
-                """)
+            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")
     
-    # Footer with enhanced info
-    gr.Markdown("""
-    ---
-    **🔧 Technical Details:**
-    - **Model**: Qwen/Qwen2.5-Math-7B-Instruct (Specialized for Mathematics)
-    - **Features**: Real-time streaming • LaTeX rendering • Step-by-step solutions
-    
-    **💡 Usage Tips:**
-    - Be specific about what you want to solve
-    - Request step-by-step solutions explicitly  
-    - Use "lim(x->0)" for limits (mobile-friendly)
-    - Crank tokens to 1500+ for complex problems
-    """)
+    # 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"]
+        ],
+        inputs=msg,
+        label="💡 Quick Examples"
+    )
     
     # Event handlers
-    def handle_sample(difficulty):
-        sample = get_random_sample()
-        return sample
+    def submit_message(message, history):
+        return chat_response(message, history)
+    
+    def clear_chat():
+        return [], ""
+    
+    msg.submit(submit_message, [msg, chatbot], [msg, chatbot])
+    submit_btn.click(submit_message, [msg, chatbot], [msg, chatbot])
     
     sample_btn.click(
-        handle_sample,
+        insert_sample_to_chat,
         inputs=[difficulty_preset],
-        outputs=gr.Textbox(visible=False)  # This will trigger the chat input
+        outputs=msg
     )
     
     help_btn.click(
         show_help,
-        outputs=gr.Markdown(label="Help", visible=True)
+        outputs=gr.Markdown(visible=True, label="Help")
     )
+    
+    # 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()