Update app.py

app.py

@@ -1,11 +1,12 @@
 import gradio as gr
 import os
 import re
-from openai import OpenAI
 import time
+from openai import OpenAI
+from together import Together
 
-# Function to generate math solution using the Phi-4-reasoning-plus model
-def generate_math_solution(api_key, problem_text, history=None):
+# Function to generate math solution using the Phi-4-reasoning-plus model via OpenRouter
+def generate_math_solution_openrouter(api_key, problem_text, history=None):
     if not api_key.strip():
         return "Please enter your OpenRouter API key.", history
     
@@ -65,13 +66,89 @@ def generate_math_solution(api_key, problem_text, history=None):
         error_message = f"Error: {str(e)}"
         return error_message, history
 
+# Function to generate math solution using Together AI with support for images
+def generate_math_solution_together(api_key, problem_text, image=None, history=None):
+    if not api_key.strip():
+        return "Please enter your Together AI API key.", history
+    
+    if not problem_text.strip() and image is None:
+        return "Please enter a math problem or upload an image of a math problem.", history
+    
+    try:
+        client = Together(api_key=api_key)
+        
+        system_message = {
+            "role": "system",
+            "content": """You are an expert math tutor who explains concepts clearly and thoroughly.
+             Analyze the given math problem and provide a detailed step-by-step solution.
+             For each step:
+             1. Show the mathematical operation
+             2. Explain why this step is necessary
+             3. Connect it to relevant mathematical concepts
+             
+             Format your response with clear section headers using markdown.
+             Begin with an "Initial Analysis" section, follow with numbered steps,
+             and conclude with a "Final Answer" section."""
+        }
+        
+        # Start with text content
+        user_content = [{"type": "text", "text": f"Solve this math problem: {problem_text}"}]
+        
+        # Add image if provided
+        if image is not None:
+            user_content.append({
+                "type": "image_url",
+                "image_url": {"url": image}
+            })
+        
+        messages = [system_message]
+        
+        # Add conversation history if it exists
+        if history:
+            for exchange in history:
+                # For simplicity, we're assuming text-only history here
+                messages.append({"role": "user", "content": exchange[0]})
+                if exchange[1]:  # Check if there's a response
+                    messages.append({"role": "assistant", "content": exchange[1]})
+        
+        # Add the current problem
+        messages.append({"role": "user", "content": user_content})
+        
+        # Create the completion
+        response = client.chat.completions.create(
+            model="meta-llama/Llama-Vision-Free",
+            messages=messages,
+            stream=False
+        )
+        
+        solution = response.choices[0].message.content
+        
+        # Update history - for simplicity, just store the text problem
+        if history is None:
+            history = []
+        history.append((problem_text, solution))
+        
+        return solution, history
+        
+    except Exception as e:
+        error_message = f"Error: {str(e)}"
+        return error_message, history
+
 # Function to verify API key format
-def validate_api_key(api_key):
+def validate_openrouter_key(api_key):
     # This is a simple check - OpenRouter keys typically start with "sk-or-"
     if api_key.startswith("sk-or-") and len(api_key) > 20:
         return True
     return False
 
+# Function to validate Together AI API key (basic format check)
+def validate_together_key(api_key):
+    # Together API keys usually have a specific format
+    # This is just a simple length check for now
+    if len(api_key) > 20:
+        return True
+    return False
+
 # Function to process LaTeX in the solution
 def process_solution(solution):
     # Replace $...$ with $$...$$ for better rendering in Gradio markdown
@@ -83,21 +160,23 @@ def create_demo():
     with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as demo:
         gr.Markdown("# 📚 Advanced Math Tutor")
         gr.Markdown("""
-        This application uses Microsoft's Phi-4-reasoning-plus model to provide step-by-step solutions 
-        to math problems. Enter your math problem, and get a detailed explanation with clear reasoning steps.
+        This application provides step-by-step solutions to math problems using advanced AI models.
+        Choose between OpenRouter's Phi-4-reasoning-plus for text-based problems or Together AI's 
+        Llama-Vision for problems with images.
         """)
         
         # Main tabs
         with gr.Tabs():
-            with gr.TabItem("Problem Solver"):
+            # Text-based problem solver (OpenRouter)
+            with gr.TabItem("Text Problem Solver (OpenRouter)"):
                 with gr.Row():
                     with gr.Column(scale=1):
-                        api_key_input = gr.Textbox(
+                        openrouter_api_key = gr.Textbox(
                             label="OpenRouter API Key", 
                             placeholder="Enter your OpenRouter API key (starts with sk-or-)",
                             type="password"
                         )
-                        problem_input = gr.Textbox(
+                        text_problem_input = gr.Textbox(
                             label="Math Problem", 
                             placeholder="Enter your math problem here...",
                             lines=5
@@ -109,46 +188,97 @@ def create_demo():
                                 ["Calculate the area of a circle with radius 5 cm"],
                                 ["Find all values of x that satisfy the equation: log₂(x-1) + log₂(x+3) = 5"]
                             ],
-                            inputs=[problem_input],
+                            inputs=[text_problem_input],
                             label="Example Problems"
                         )
                         with gr.Row():
-                            submit_btn = gr.Button("Solve Problem", variant="primary")
-                            clear_btn = gr.Button("Clear")
+                            openrouter_submit_btn = gr.Button("Solve Problem", variant="primary")
+                            openrouter_clear_btn = gr.Button("Clear")
                     
                 with gr.Column(scale=2):
-                    solution_output = gr.Markdown(label="Solution")
+                    openrouter_solution_output = gr.Markdown(label="Solution")
                 
                 # Store conversation history (invisible to user)
-                conversation_history = gr.State(value=None)
+                openrouter_conversation_history = gr.State(value=None)
                 
                 # Button actions
-                submit_btn.click(
-                    fn=generate_math_solution,
-                    inputs=[api_key_input, problem_input, conversation_history],
-                    outputs=[solution_output, conversation_history]
+                openrouter_submit_btn.click(
+                    fn=generate_math_solution_openrouter,
+                    inputs=[openrouter_api_key, text_problem_input, openrouter_conversation_history],
+                    outputs=[openrouter_solution_output, openrouter_conversation_history]
                 )
                 
-                clear_btn.click(
+                openrouter_clear_btn.click(
                     fn=lambda: ("", None),
                     inputs=[],
-                    outputs=[solution_output, conversation_history]
+                    outputs=[openrouter_solution_output, openrouter_conversation_history]
                 )
             
+            # Image-based problem solver (Together AI)
+            with gr.TabItem("Image Problem Solver (Together AI)"):
+                with gr.Row():
+                    with gr.Column(scale=1):
+                        together_api_key = gr.Textbox(
+                            label="Together AI API Key", 
+                            placeholder="Enter your Together AI API key",
+                            type="password"
+                        )
+                        together_problem_input = gr.Textbox(
+                            label="Problem Description (Optional)", 
+                            placeholder="Enter additional context for the image problem...",
+                            lines=3
+                        )
+                        together_image_input = gr.Image(
+                            label="Upload Math Problem Image",
+                            type="filepath"
+                        )
+                        with gr.Row():
+                            together_submit_btn = gr.Button("Solve Problem", variant="primary")
+                            together_clear_btn = gr.Button("Clear")
+                    
+                with gr.Column(scale=2):
+                    together_solution_output = gr.Markdown(label="Solution")
+                
+                # Store conversation history (invisible to user)
+                together_conversation_history = gr.State(value=None)
+                
+                # Button actions
+                together_submit_btn.click(
+                    fn=generate_math_solution_together,
+                    inputs=[together_api_key, together_problem_input, together_image_input, together_conversation_history],
+                    outputs=[together_solution_output, together_conversation_history]
+                )
+                
+                together_clear_btn.click(
+                    fn=lambda: ("", None),
+                    inputs=[],
+                    outputs=[together_solution_output, together_conversation_history]
+                )
+            
+            # Help tab
             with gr.TabItem("Help"):
                 gr.Markdown("""
                 ## How to Use the Advanced Math Tutor
                 
                 ### Getting Started
-                1. You'll need an API key from OpenRouter to use this app
+                
+                #### For Text-Based Problems (OpenRouter)
+                1. You'll need an API key from OpenRouter
                 2. Sign up at [OpenRouter](https://openrouter.ai/) to get your API key
-                3. Enter your API key in the designated field
+                3. Enter your API key in the designated field in the "Text Problem Solver" tab
+                
+                #### For Image-Based Problems (Together AI)
+                1. You'll need an API key from Together AI
+                2. Sign up at [Together AI](https://www.together.ai/) to get your API key
+                3. Enter your API key in the designated field in the "Image Problem Solver" tab
+                4. Upload an image of your math problem
+                5. Optionally add text to provide additional context
                 
                 ### Solving Math Problems
-                - Type or paste your math problem in the input field
+                - For text problems: Type or paste your math problem in the input field
+                - For image problems: Upload a clear image of the math problem
                 - Click "Solve Problem" to get a detailed step-by-step solution
                 - The solution will include explanations for each step
-                - You can also try one of the provided example problems
                 
                 ### Tips for Best Results
                 - Be specific in your problem description
@@ -156,6 +286,7 @@ def create_demo():
                 - For complex equations, use clear notation
                 - For algebraic expressions, use ^ for exponents (e.g., x^2 for x²)
                 - Use parentheses to group terms clearly
+                - For images, ensure the math problem is clearly visible and well-lit
                 
                 ### Types of Problems You Can Solve
                 - Algebra (equations, inequalities, systems of equations)
@@ -171,8 +302,9 @@ def create_demo():
         gr.Markdown("""
         ---
         ### About
-        This application uses Microsoft's Phi-4-reasoning-plus model via OpenRouter to generate step-by-step solutions. 
-        Your API key is required but not stored permanently.
+        This application uses Microsoft's Phi-4-reasoning-plus model via OpenRouter for text-based problems
+        and Llama-Vision-Free via Together AI for image-based problems. 
+        Your API keys are required but not stored permanently.
         """)
     
     return demo