Update app. py

app. py

@@ -0,0 +1,258 @@
+import gradio as gr
+import requests
+import base64
+from io import BytesIO
+from PIL import Image
+import json
+
+# Hugging Face API configuration
+API_URL = "https://api.anthropic.com/v1/messages"
+
+def encode_image(image):
+    """Convert PIL Image to base64 string"""
+    buffered = BytesIO()
+    image.save(buffered, format="PNG")
+    img_str = base64.b64encode(buffered.getvalue()).decode()
+    return img_str
+
+def analyze_math_work(image, grade_level, topic):
+    """
+    Analyze student's math work using Claude API
+    """
+    try:
+        # Encode the image
+        img_base64 = encode_image(image)
+        
+        # Construct the prompt based on grade level and topic
+        system_prompt = f"""You are a patient and encouraging mathematics teacher for {grade_level} students. 
+Your role is to:
+1. Carefully examine the student's written work in the image
+2. Identify what mathematical problem they are solving
+3. Check their working step-by-step
+4. Point out any errors with clear, kind explanations
+5. Provide the correct solution with detailed steps
+6. Encourage the student and give tips for improvement
+
+Focus on {topic} topics appropriate for {grade_level} level.
+Be specific about where mistakes occur and explain why they are mistakes.
+Always be encouraging and supportive."""
+
+        user_prompt = """Please analyze the mathematical work shown in this image. 
+
+Provide your feedback in the following format:
+
+**Problem Identified:**
+[State what problem the student is trying to solve]
+
+**Student's Approach:**
+[Describe what the student did]
+
+**Evaluation:**
+[Is it correct? If not, where did they go wrong?]
+
+**Detailed Feedback:**
+[Step-by-step explanation of mistakes and corrections]
+
+**Correct Solution:**
+[Show the complete correct solution with all steps]
+
+**Encouragement & Tips:**
+[Positive feedback and helpful tips]"""
+
+        # Make API request
+        response = requests.post(
+            API_URL,
+            headers={
+                "Content-Type": "application/json",
+            },
+            json={
+                "model": "claude-sonnet-4-20250514",
+                "max_tokens": 2000,
+                "system": system_prompt,
+                "messages": [
+                    {
+                        "role": "user",
+                        "content": [
+                            {
+                                "type": "image",
+                                "source": {
+                                    "type": "base64",
+                                    "media_type": "image/png",
+                                    "data": img_base64
+                                }
+                            },
+                            {
+                                "type": "text",
+                                "text": user_prompt
+                            }
+                        ]
+                    }
+                ]
+            },
+            timeout=60
+        )
+        
+        if response.status_code == 200:
+            result = response.json()
+            feedback = result['content'][0]['text']
+            return feedback
+        else:
+            return f"Error: API returned status code {response.status_code}\n{response.text}"
+            
+    except Exception as e:
+        return f"Error analyzing work: {str(e)}\n\nPlease make sure:\n1. Your image is clear and readable\n2. The mathematical work is visible\n3. Try again or contact support if the issue persists"
+
+def create_practice_problem(grade_level, topic):
+    """Generate a practice problem for the student"""
+    try:
+        prompt = f"""Generate a {grade_level} level mathematics practice problem about {topic}.
+
+Provide:
+1. A clear problem statement
+2. Any necessary context or setup
+3. Space for the student to work (indicate this)
+
+Make it appropriate for {grade_level} students and engaging."""
+
+        response = requests.post(
+            API_URL,
+            headers={
+                "Content-Type": "application/json",
+            },
+            json={
+                "model": "claude-sonnet-4-20250514",
+                "max_tokens": 1000,
+                "messages": [
+                    {
+                        "role": "user",
+                        "content": prompt
+                    }
+                ]
+            },
+            timeout=30
+        )
+        
+        if response.status_code == 200:
+            result = response.json()
+            return result['content'][0]['text']
+        else:
+            return "Error generating practice problem. Please try again."
+            
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+# Create the Gradio interface
+with gr.Blocks(title="Math Learning Assistant", theme=gr.themes.Soft()) as app:
+    gr.Markdown("""
+    # 📚 Math Learning Assistant
+    ### Learn mathematics by practicing and getting instant feedback!
+    
+    **How to use:**
+    1. Choose your grade level and math topic
+    2. Either draw your solution using the canvas OR upload a photo of your written work
+    3. Click "Check My Work" to get detailed feedback from your AI teacher
+    4. Learn from the feedback and try again!
+    """)
+    
+    with gr.Row():
+        with gr.Column(scale=1):
+            grade_level = gr.Dropdown(
+                choices=[
+                    "Primary 1-2 (Ages 6-8)",
+                    "Primary 3-4 (Ages 8-10)", 
+                    "Primary 5-6 (Ages 10-12)",
+                    "Secondary 1-2 (Ages 12-14)",
+                    "Secondary 3-4 (Ages 14-16)"
+                ],
+                label="Select Your Grade Level",
+                value="Primary 3-4 (Ages 8-10)"
+            )
+            
+            topic = gr.Dropdown(
+                choices=[
+                    "Addition & Subtraction",
+                    "Multiplication & Division",
+                    "Fractions",
+                    "Decimals",
+                    "Percentages",
+                    "Algebra (Basic)",
+                    "Algebra (Advanced)",
+                    "Geometry",
+                    "Word Problems",
+                    "Equations",
+                    "Ratios & Proportions",
+                    "Measurements"
+                ],
+                label="Select Math Topic",
+                value="Addition & Subtraction"
+            )
+            
+            get_problem_btn = gr.Button("📝 Get Practice Problem", variant="secondary")
+            practice_problem = gr.Textbox(
+                label="Practice Problem",
+                placeholder="Click 'Get Practice Problem' to receive a new problem to solve",
+                lines=6
+            )
+    
+    with gr.Row():
+        with gr.Column(scale=1):
+            gr.Markdown("### ✍️ Write or Upload Your Solution")
+            
+            with gr.Tabs():
+                with gr.Tab("Draw Your Work"):
+                    sketchpad = gr.Sketchpad(
+                        label="Write your solution here (use your finger or mouse)",
+                        type="pil",
+                        brush=gr.Brush(colors=["#000000", "#0000FF", "#FF0000"], default_size=3)
+                    )
+                    
+                with gr.Tab("Upload Photo"):
+                    image_upload = gr.Image(
+                        label="Upload a photo of your written work",
+                        type="pil"
+                    )
+            
+            check_btn = gr.Button("🔍 Check My Work", variant="primary", size="lg")
+        
+        with gr.Column(scale=1):
+            gr.Markdown("### 📊 Teacher's Feedback")
+            feedback_output = gr.Textbox(
+                label="Detailed Feedback",
+                lines=20,
+                placeholder="Your feedback will appear here after clicking 'Check My Work'"
+            )
+    
+    gr.Markdown("""
+    ---
+    ### 💡 Tips for Best Results:
+    - Write clearly and show all your steps
+    - If uploading a photo, make sure it's well-lit and in focus
+    - Include the problem you're solving at the top
+    - Don't worry about mistakes - they help you learn!
+    """)
+    
+    # Event handlers
+    def check_work(sketch, upload, grade, topic_val):
+        # Use whichever image is available (sketchpad takes priority)
+        image_to_analyze = sketch if sketch is not None else upload
+        
+        if image_to_analyze is None:
+            return "⚠️ Please draw your solution or upload an image of your work first!"
+        
+        return analyze_math_work(image_to_analyze, grade, topic_val)
+    
+    check_btn.click(
+        fn=check_work,
+        inputs=[sketchpad, image_upload, grade_level, topic],
+        outputs=feedback_output
+    )
+    
+    get_problem_btn.click(
+        fn=create_practice_problem,
+        inputs=[grade_level, topic],
+        outputs=practice_problem
+    )
+
+# Launch the app
+if __name__ == "__main__":
+    app.launch(share=True)