Update app.py

app.py

@@ -4,18 +4,19 @@ from huggingface_hub import InferenceClient
 from datasets import load_dataset
 import random
 import re
+
 # Load math datasets for sample problems
 fw = load_dataset("HuggingFaceFW/fineweb-edu", name="sample-10BT", split="train", streaming=True)
 ds = load_dataset("HuggingFaceH4/ultrachat_200k", streaming=True)
+
 def load_sample_problems():
     """Load sample problems from math datasets"""
     try:
-        # Load GSM8K for sample problems
         gsm8k = load_dataset("openai/gsm8k", "main", streaming=True)
         samples = []
         for i, item in enumerate(gsm8k["train"]):
             samples.append(item["question"])
-            if i >= 50:  # Load 50 samples
+            if i >= 50:
                 break
         return samples
     except:
@@ -27,7 +28,6 @@ def load_sample_problems():
             "Solve the system: x + 2y = 7, 3x - y = 4"
         ]
 
-# Initialize sample problems
 math_samples = load_sample_problems()
 
 def create_math_system_message():
@@ -55,41 +55,34 @@ def create_math_system_message():
 5. **Alternative Methods** - Mention other possible approaches
 
 Always be precise, educational, and encourage mathematical thinking."""
-def respond(
-    message,
-    history: list[dict[str, str]],
-    system_message,
-    max_tokens,
-    temperature,
-    top_p,
-):
-    def render_latex(text):
-    # Wrap inline math properly
-    text = re.sub(r'\$(.*?)\$', r'$$\1$$', text)  # Display mode
+
+def render_latex(text):
+    """Format math expressions for LaTeX rendering"""
+    if not text:  # Safety check
+        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\to0}$
+    text = re.sub(r'lim\(([a-z])→([0-9.]+)\)', r'$\lim_{ \1 \to \2}$', text)
     return text
 
-    yield render_latex(response)
-    
-    # Show initial thinking message
+def respond(message, history, system_message, max_tokens, temperature, top_p):
+    # Show initial thinking
     yield "🤔 Thinking for the suitable answer..."
     
-    """
-    Enhanced response function for mathematical problem solving
-    """
-    # Use Qwen Math model for better mathematical reasoning
-    client = InferenceClient(
-        model="Qwen/Qwen2.5-Math-7B-Instruct"  # Specialized math model
-    )
-
-    # Build message history
+    # Use Qwen Math model
+    client = InferenceClient(model="Qwen/Qwen2.5-Math-7B-Instruct")
+    
+    # Build messages
     messages = [{"role": "system", "content": system_message}]
     messages.extend(history)
     messages.append({"role": "user", "content": message})
-
+    
     response = ""
-    # Add this line to ensure no truncation
-    max_tokens = max(max_tokens, 1536)  # Floor at 1536 for complex math
-
+    max_tokens = max(max_tokens, 1536)  # Ensure no truncation
+    
     try:
         for message_chunk in client.chat_completion(
             messages,
@@ -99,12 +92,19 @@ def respond(
             top_p=top_p,
         ):
             choices = message_chunk.choices
-            token = ""
             if len(choices) and choices[0].delta.content:
                 token = choices[0].delta.content
-
-            response += token
-            yield response
+                response += token
+                # Stream raw during generation, format at end
+                if "```" in response or len(response) > 500:  # Format after substantial response
+                    formatted = render_latex(response)
+                    yield formatted
+                else:
+                    yield response
+                    
+        # Final formatted response
+        final_formatted = render_latex(response)
+        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"
@@ -116,17 +116,22 @@ def get_random_sample():
         return random.choice(math_samples)
     return "Solve for x: 2x² + 5x - 3 = 0"
 
-def set_difficulty_preset(difficulty):
-    """Set temperature and other params based on difficulty"""
-    presets = {
-        "Elementary": (0.2, 0.8, 512),    # Low temp for accuracy
-        "High School": (0.3, 0.85, 768),  # Balanced
-        "College": (0.4, 0.9, 1024),      # More exploration
-        "Competition": (0.3, 0.8, 1536)   # Detailed solutions
-    }
-    return presets.get(difficulty, (0.3, 0.85, 768))
+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
 
-# Create the enhanced ChatInterface
+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?"
+    """
+
+# Create ChatInterface
 chatbot = gr.ChatInterface(
     respond,
     type="messages",
@@ -144,79 +149,40 @@ chatbot = gr.ChatInterface(
             lines=3,
             max_lines=10
         ),
-        gr.Slider(
-            minimum=256, 
-            maximum=2048, 
-            value=768, 
-            step=64, 
-            label="📝 Max Tokens (Solution Length)"
-        ),
-        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)",
-        ),
+        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"),
     ],
     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?"],
-        ["If log₂(x) + log₂(x+6) = 4, find the value of x."],
-        ["A bag contains 5 red balls and 7 blue balls. What's the probability of drawing 2 red balls without replacement?"],
         ["Find the limit: lim(x→0) (sin(x)/x)"],
-        ["Solve the system of equations: x + 2y = 7 and 3x - y = 4"],
-        ["What is the integral of ∫(2x³ - 5x + 3)dx?"],
-        ["In a right triangle, if one angle is 30° and the hypotenuse is 10, find the lengths of the other two sides."]
+        ["Solve the system: x + 2y = 7, 3x - y = 4"],
+        ["What is the integral of ∫(2x³ - 5x + 3)dx?"]
     ],
-    cache_examples=False,  # Don't cache for dynamic math problems
+    cache_examples=False,
     concurrency_limit=10,
 )
 
-# Enhanced interface with additional features
+# Main interface
 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;
-    }
+    .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; }
     """
 ) as demo:
     
-    gr.Markdown("""
-    # 🧮 **Mathetics AI** - Advanced Mathematics Solver
+    gr.Markdown("# 🧮 **Mathematics AI** - Advanced Mathematics Solver\n**Your Personal AI Math Tutor**")
     
-    **Your Personal AI Math Tutor** | Specialized in step-by-step problem solving across all mathematical domains
-    """)
-
-
     with gr.Row():
         with gr.Column(scale=4):
-            # Main chat interface
             chatbot.render()
             
         with gr.Column(scale=1):
-            # Sidebar with additional features
             with gr.Accordion("🎲 **Quick Actions**", open=True):
-                
                 difficulty_preset = gr.Dropdown(
                     choices=["Elementary", "High School", "College", "Competition"],
                     value="High School",
@@ -224,54 +190,29 @@ with gr.Blocks(
                     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")
+                sample_btn = gr.Button("🎯 Get Sample Problem", variant="secondary")
+                help_btn = gr.Button("❓ Math Help Tips", variant="secondary")
                 
-                # Quick math tools
                 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
-                """)
+                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")
     
-    # Footer with information
     gr.Markdown("""
     ---
-    **🔧 Technical Details:**
-    - **Model**: Qwen/Qwen2.5-Math-7B-Instruct (Specialized for Mathematics)
-    - **Features**: Real-time streaming responses, step-by-step solutions
+    **🔧 Technical Details:** Qwen/Qwen2.5-Math-7B-Instruct | Real-time streaming responses
     
-    **💡 Usage Tips:**
-    - Be specific about what you want to find or solve
-    - Mention if you want step-by-step solutions
-    - Ask for alternative solution methods
-    - Request verification of your own solutions
+    **💡 Usage Tips:** Be specific • Request step-by-step • Ask for verification
     """)
-        
     
-    # Event handlers for additional functionality
-    def insert_sample(difficulty):
-        sample = get_random_sample()
-        return sample
-    
-    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?"
-        """
-    
-    # Connect sample button (this would need to be integrated with the chat properly)
+    # Event handlers
     sample_btn.click(
-        lambda x: get_random_sample(),
+        insert_sample_to_chat,
         inputs=[difficulty_preset],
-        outputs=[]  # Would need proper integration with chat input
+        outputs=[chatbot]  # Actually feeds into chat now!
+    )
+    
+    help_btn.click(
+        show_help,
+        outputs=gr.Markdown()  # Could show in a separate output
     )
 
 if __name__ == "__main__":