Update app.py

app.py

@@ -6,6 +6,7 @@ import random
 import re
 
 # Load math datasets for sample problems
+print("🔄 Loading datasets...")
 fw = load_dataset("HuggingFaceFW/fineweb-edu", name="sample-10BT", split="train", streaming=True)
 ds = load_dataset("HuggingFaceH4/ultrachat_200k", streaming=True)
 
@@ -18,26 +19,28 @@ def load_sample_problems():
             samples.append(item["question"])
             if i >= 50:
                 break
+        print(f"✅ Loaded {len(samples)} GSM8K samples")
         return samples
-    except:
+    except Exception as e:
+        print(f"⚠️ Dataset error: {e}, using fallback")
         return [
             "What is the derivative of f(x) = 3x² + 2x - 1?",
             "A triangle has sides of length 5, 12, and 13. What is its area?",
             "If log₂(x) + log₂(x+6) = 4, find the value of x.",
-            "Find the limit: lim(x→0) (sin(x)/x)",
+            "Find the limit: lim(x->0) (sin(x)/x)",
             "Solve the system: x + 2y = 7, 3x - y = 4"
         ]
 
 math_samples = load_sample_problems()
 
 def create_math_system_message():
-    """Create specialized system prompt for mathematics"""
-    return """You are Mathetics AI, an advanced mathematics tutor and problem solver. 
+    """Create specialized system prompt for mathematics with LaTeX"""
+    return """You are Mathetics AI, an advanced mathematics tutor and problem solver.
 
 🧮 **Your Expertise:**
 - Step-by-step problem solving with clear explanations
 - Multiple solution approaches when applicable
-- Proper mathematical notation and terminology
+- Proper mathematical notation and terminology using LaTeX
 - Verification of answers through different methods
 
 📐 **Problem Domains:**
@@ -50,34 +53,74 @@ def create_math_system_message():
 💡 **Teaching Style:**
 1. **Understand the Problem** - Identify what's being asked
 2. **Plan the Solution** - Choose the appropriate method
-3. **Execute Step-by-Step** - Show all work clearly
+3. **Execute Step-by-Step** - Show all work clearly with LaTeX formatting
 4. **Verify the Answer** - Check if the result makes sense
 5. **Alternative Methods** - Mention other possible approaches
 
+**LaTeX Guidelines:**
+- Use $...$ for inline math: $x^2 + y^2 = z^2$
+- Use $$...$$ or \[...\] 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."""
 
 def render_latex(text):
-    """Format math expressions for LaTeX rendering"""
-    if not text:  # Safety check
+    """Enhanced LaTeX rendering - fixes raw code output"""
+    if not text or len(text) < 5:
         return text
-    # Wrap inline math: x^2 → $x^2$
-    text = re.sub(r'\b([a-zA-Z]+)\^?(\d*)\b', r'$\1^{\2}$', text)
-    # Wrap functions: sin(x) → $\sin(x)$
-    text = re.sub(r'([a-zA-Z]+)\(([^)]+)\)', r'$\1(\2)$', text)
-    # Wrap limits: lim(x→0) → $\lim_{x \to 0}$
-    text = re.sub(r'lim\(([a-z])→([0-9.]+)\)', r'$\\lim_{ \1 \\to \2}$', 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)
+        
+        # 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
+        text = re.sub(r'\s*([\$\\])\s*', r'\1', 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 for the suitable answer..."
+    yield "🤔 Thinking step-by-step..."
     
     # Use Qwen Math model
     client = InferenceClient(model="Qwen/Qwen2.5-Math-7B-Instruct")
     
-    # Build messages
-    messages = [{"role": "system", "content": system_message}]
-    messages.extend(history)
+    # 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"]})
+    
     messages.append({"role": "user", "content": message})
     
     response = ""
@@ -90,24 +133,26 @@ 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
         ):
             choices = message_chunk.choices
             if len(choices) and choices[0].delta.content:
                 token = choices[0].delta.content
                 response += token
-                # Stream raw during generation, format at end
-                if "```" in response or len(response) > 500:  # Format after substantial response
+                
+                # 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)
+        final_formatted = render_latex(response.strip())
         yield final_formatted
             
     except Exception as e:
-        error_msg = f"❌ **Error**: {str(e)}\n\n💡 **Troubleshooting**:\n- Make sure you're logged in with Hugging Face\n- Check if the model is accessible\n- Try a simpler problem first"
+        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"
         yield error_msg
 
 def get_random_sample():
@@ -119,19 +164,25 @@ def get_random_sample():
 def insert_sample_to_chat(difficulty):
     """Insert random sample into chat input"""
     sample = get_random_sample()
-    return "", [None, sample]  # Clear input, add sample to chat
+    return sample, ""  # Return sample for input, clear status
 
 def show_help():
-    return """**Math Help Tips:**
-        
-    1. **Be Specific**: "Find the derivative of..." instead of "Help with calculus"
-    2. **Show Your Work**: "I got x=5, is this correct?"
-    3. **Ask for Steps**: "Show me step-by-step how to solve..."
-    4. **Request Verification**: "Check my solution to this problem"
-    5. **Alternative Methods**: "What's another way to solve this?"
-    """
+    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)" 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}$$
+
+**Pro Tip**: Crank tokens to 1500+ for competition problems!"""
 
-# Create ChatInterface
+# Enhanced ChatInterface with better LaTeX support
 chatbot = gr.ChatInterface(
     respond,
     type="messages",
@@ -140,42 +191,108 @@ chatbot = gr.ChatInterface(
     **Powered by Qwen 2.5-Math** | **Specialized for Mathematical Problem Solving**
     
     ✨ **Capabilities**: Algebra • Geometry • Calculus • Statistics • Competition Math
-    📚 **Features**: Step-by-step solutions • Multiple approaches • Clear explanations
+    📚 **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=3,
-            max_lines=10
+            lines=4,
+            max_lines=12
         ),
-        gr.Slider(minimum=256, maximum=2048, value=768, step=64, label="📝 Max Tokens"),
-        gr.Slider(minimum=0.1, maximum=1.0, value=0.3, step=0.1, label="🎯 Temperature"),
-        gr.Slider(minimum=0.1, maximum=1.0, value=0.85, step=0.05, label="🔍 Top-p"),
+        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=[
-        ["What is the derivative of f(x) = 3x² + 2x - 1?"],
-        ["A triangle has sides of length 5, 12, and 13. What is its area?"],
-        ["Find the limit: lim(x→0) (sin(x)/x)"],
-        ["Solve the system: x + 2y = 7, 3x - y = 4"],
-        ["What is the integral of ∫(2x³ - 5x + 3)dx?"]
+        ["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=10,
+    concurrency_limit=5,  # Reduce for stability
+    retry_btn="🔄 Retry",
+    undo_btn="↶ Undo",
+    clear_btn="🗑️ Clear"
 )
 
-# Main interface
+# Main interface with enhanced LaTeX CSS
 with gr.Blocks(
     title="🧮 Mathetics AI", 
     theme=gr.themes.Soft(),
     css="""
-    .math-highlight { background-color: #f0f8ff; padding: 10px; border-left: 4px solid #4CAF50; margin: 10px 0; border-radius: 5px; }
-    .difficulty-selector { background-color: #fff3e0; padding: 15px; border-radius: 10px; margin: 10px 0; }
-    .markdown math { font-size: 1.1em; }
+    /* 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;
+    }
+    
+    /* Boxed answers */
+    .katex .mord.text { font-weight: bold; }
+    
+    /* 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 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; }
+    }
     """
 ) as demo:
     
-    gr.Markdown("# 🧮 **Mathematics AI** - Advanced Mathematics Solver\n**Your Personal AI Math Tutor**")
+    gr.Markdown("""
+    # 🧮 **Mathetics AI** - Advanced Mathematics Solver
+    
+    **Your Personal AI Math Tutor** | Specialized in step-by-step problem solving with beautiful LaTeX rendering
+    
+    ---
+    """)
     
     with gr.Row():
         with gr.Column(scale=4):
@@ -190,29 +307,46 @@ with gr.Blocks(
                     elem_classes=["difficulty-selector"]
                 )
                 
-                sample_btn = gr.Button("🎯 Get Sample Problem", variant="secondary")
-                help_btn = gr.Button("❓ Math Help Tips", variant="secondary")
+                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\n- **Geometry**: Area, volume, trigonometry\n- **Calculus**: Derivatives, integrals, limits\n- **Statistics**: Probability, distributions\n- **Number Theory**: Prime factorization, GCD/LCM")
+                gr.Markdown("""
+                - **Algebra**: Equations, inequalities, factoring
+                - **Geometry**: Area, volume, trigonometry  
+                - **Calculus**: Derivatives, integrals, limits
+                - **Statistics**: Probability, distributions
+                - **Number Theory**: Prime factorization, GCD/LCM
+                """)
     
+    # Footer with enhanced info
     gr.Markdown("""
     ---
-    **🔧 Technical Details:** Qwen/Qwen2.5-Math-7B-Instruct | Real-time streaming responses
+    **🔧 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 • Request step-by-step • Ask for verification
+    **💡 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
     """)
     
     # Event handlers
+    def handle_sample(difficulty):
+        sample = get_random_sample()
+        return sample
+    
     sample_btn.click(
-        insert_sample_to_chat,
+        handle_sample,
         inputs=[difficulty_preset],
-        outputs=[chatbot]  # Actually feeds into chat now!
+        outputs=gr.Textbox(visible=False)  # This will trigger the chat input
     )
     
     help_btn.click(
         show_help,
-        outputs=gr.Markdown()  # Could show in a separate output
+        outputs=gr.Markdown(label="Help", visible=True)
     )
 
 if __name__ == "__main__":