Math Solution - Advanced Math Tutor

import gradio as gr import os import re import time import base64 from openai import OpenAI from together import Together from PIL import Image import io import markdown from datetime import datetime import tempfile import weasyprint from pathlib import Path # Function to convert markdown to HTML with styling and MathJax def markdown_to_html(markdown_text, problem_text="", include_problem=True): """Convert markdown to styled HTML with proper math rendering""" # Convert markdown to HTML html_content = markdown.markdown(markdown_text, extensions=['tables', 'fenced_code']) # Replace LaTeX delimiters for MathJax # Handle display math: \[ ... \] and $$ ... $$ html_content = re.sub(r'\\\[(.*?)\\\]', r'\\[\1\\]', html_content, flags=re.DOTALL) html_content = re.sub(r'\$\$(.*?)\$\$', r'\\[\1\\]', html_content, flags=re.DOTALL) # Handle inline math: $ ... $ and $ ... $ html_content = re.sub(r'\\$(.*?)\\$', r'\$\1\$', html_content) # Be careful with single $ as it might be regular dollar signs html_content = re.sub(r'(? Math Solution - Advanced Math Tutor

{f'''

📝 Problem Statement

{problem_text}

''' if include_problem and problem_text.strip() else ''}

🔍 Solution

{html_content}

Generated on: {timestamp}

""" return styled_html # Function to save HTML to file def save_html_to_file(html_content, filename_prefix="math_solution"): """Save HTML content to a temporary file and return the file path""" timestamp = datetime.now().strftime("%Y%m%d_%H%M%S") filename = f"{filename_prefix}_{timestamp}.html" # Create a temporary file temp_dir = tempfile.gettempdir() file_path = os.path.join(temp_dir, filename) with open(file_path, 'w', encoding='utf-8') as f: f.write(html_content) return file_path # Function to convert HTML to PDF with proper math rendering def html_to_pdf(html_content, filename_prefix="math_solution"): """Convert HTML content to PDF and return the file path""" timestamp = datetime.now().strftime("%Y%m%d_%H%M%S") filename = f"{filename_prefix}_{timestamp}.pdf" # Create a temporary file for PDF temp_dir = tempfile.gettempdir() pdf_path = os.path.join(temp_dir, filename) try: # Note: WeasyPrint doesn't support MathJax rendering # For PDF with proper math, we need to use a different approach # This creates a basic PDF - for proper math rendering in PDF, # consider using tools like puppeteer or selenium with headless browser weasyprint.HTML(string=html_content).write_pdf(pdf_path) return pdf_path except Exception as e: print(f"Error converting to PDF: {str(e)}") return None # Enhanced function to generate math solution using OpenRouter with HTML output def generate_math_solution_openrouter(api_key, problem_text, history=None): if not api_key.strip(): return "Please enter your OpenRouter API key.", None, None, history if not problem_text.strip(): return "Please enter a math problem.", None, None, history try: client = OpenAI( base_url="https://openrouter.ai/api/v1", api_key=api_key, ) messages = [ {"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 using proper LaTeX notation wrapped in \\[ \\] for display math or \$ \$ for inline math 2. Explain why this step is necessary 3. Connect it to relevant mathematical concepts IMPORTANT: Use proper LaTeX notation for all mathematical expressions: - Use \\[ ... \\] for display (centered) equations - Use \$ ... \$ for inline math - Use \\boxed{} for final answers - Use \\text{} for text within math mode - Example: \\[ A = \\pi r^{2} \\] - Example: The area \$ A \$ equals \$ \\pi r^{2} \$ Format your response using markdown with clear section headers. Begin with an "Initial Analysis" section, follow with numbered steps, and conclude with a "Final Answer" section. Use proper markdown formatting including: - Headers (##, ###) - **Bold text** for important points - Lists and numbered steps - Tables if needed for comparisons or data"""}, ] # Add conversation history if it exists if history: for exchange in history: messages.append({"role": "user", "content": exchange[0]}) if len(exchange) > 1 and exchange[1]: messages.append({"role": "assistant", "content": exchange[1]}) # Add the current problem messages.append({"role": "user", "content": f"Solve this math problem step-by-step: {problem_text}"}) # Create the completion completion = client.chat.completions.create( model="meta-llama/llama-3.3-70b-instruct:free", messages=messages, extra_headers={ "HTTP-Referer": "https://advancedmathtutor.edu", "X-Title": "Advanced Math Tutor", } ) markdown_solution = completion.choices[0].message.content # Convert to HTML html_solution = markdown_to_html(markdown_solution, problem_text) # Save HTML file html_file_path = save_html_to_file(html_solution, "openrouter_solution") # Convert to PDF pdf_file_path = html_to_pdf(html_solution, "openrouter_solution") # Update history if history is None: history = [] history.append((problem_text, markdown_solution)) return html_solution, html_file_path, pdf_file_path, history except Exception as e: error_message = f"Error: {str(e)}" return error_message, None, None, history # Enhanced function to generate math solution using Together AI with HTML output def generate_math_solution_together(api_key, problem_text, image_path=None, history=None): if not api_key.strip(): return "Please enter your Together AI API key.", None, None, history if not problem_text.strip() and image_path is None: return "Please enter a math problem or upload an image of a math problem.", None, None, history try: client = Together(api_key=api_key) # Create the base message structure messages = [ { "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 using proper LaTeX notation wrapped in \\[ \\] for display math or \$ \$ for inline math 2. Explain why this step is necessary 3. Connect it to relevant mathematical concepts IMPORTANT: Use proper LaTeX notation for all mathematical expressions: - Use \\[ ... \\] for display (centered) equations - Use \$ ... \$ for inline math - Use \\boxed{} for final answers - Use \\text{} for text within math mode - Example: \\[ A = \\pi r^{2} \\] - Example: The area \$ A \$ equals \$ \\pi r^{2} \$ Format your response using markdown with clear section headers. Begin with an "Initial Analysis" section, follow with numbered steps, and conclude with a "Final Answer" section. Use proper markdown formatting including: - Headers (##, ###) - **Bold text** for important points - Lists and numbered steps - Tables if needed for comparisons or data""" } ] # Add conversation history if it exists if history: for exchange in history: messages.append({"role": "user", "content": exchange[0]}) if len(exchange) > 1 and exchange[1]: messages.append({"role": "assistant", "content": exchange[1]}) # Prepare the user message content user_message_content = [] # Add text content if provided if problem_text.strip(): user_message_content.append({ "type": "text", "text": f"Solve this math problem: {problem_text}" }) else: user_message_content.append({ "type": "text", "text": "Solve this math problem from the image:" }) # Add image if provided if image_path: # Convert image to base64 base64_image = image_to_base64(image_path) if base64_image: user_message_content.append({ "type": "image_url", "image_url": { "url": f"data:image/jpeg;base64,{base64_image}" } }) # Add the user message with content messages.append({ "role": "user", "content": user_message_content }) # Create the completion response = client.chat.completions.create( model="ServiceNow-AI/Apriel-1.5-15b-Thinker", messages=messages, stream=False ) markdown_solution = response.choices[0].message.content # Convert to HTML problem_display = problem_text if problem_text.strip() else "Image-based problem" html_solution = markdown_to_html(markdown_solution, problem_display) # Save HTML file html_file_path = save_html_to_file(html_solution, "together_solution") # Convert to PDF pdf_file_path = html_to_pdf(html_solution, "together_solution") # Update history if history is None: history = [] history.append((problem_display, markdown_solution)) return html_solution, html_file_path, pdf_file_path, history except Exception as e: error_message = f"Error: {str(e)}" return error_message, None, None, history # Function to convert image to base64 def image_to_base64(image_path): if image_path is None: return None try: with open(image_path, "rb") as img_file: return base64.b64encode(img_file.read()).decode("utf-8") except Exception as e: print(f"Error converting image to base64: {str(e)}") return None # Define the Gradio interface def create_demo(): with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as demo: gr.Markdown("# 📚 Advanced Math Tutor") gr.Markdown(""" This application provides step-by-step solutions to math problems using advanced AI models. Solutions are generated in **HTML format** with proper mathematical notation rendering using MathJax. Choose between OpenRouter or Together AI for problem solving. """) # Main tabs with gr.Tabs(): # Text-based problem solver (OpenRouter) with gr.TabItem("Text Problem Solver (OpenRouter)"): with gr.Row(): with gr.Column(scale=1): openrouter_api_key = gr.Textbox( label="OpenRouter API Key", placeholder="Enter your OpenRouter API key (starts with sk-or-)", type="password" ) text_problem_input = gr.Textbox( label="Math Problem", placeholder="Enter your math problem here...", lines=5 ) example_problems = gr.Examples( examples=[ ["Solve the quadratic equation: 3x² + 5x - 2 = 0"], ["Find the derivative of f(x) = x³ln(x)"], ["Calculate the area of a circle with radius 5 cm"], ["Find all values of x that satisfy: log₂(x-1) + log₂(x+3) = 5"] ], inputs=[text_problem_input], label="Example Problems" ) with gr.Row(): openrouter_submit_btn = gr.Button("Solve Problem", variant="primary") openrouter_clear_btn = gr.Button("Clear") with gr.Column(scale=2): openrouter_solution_output = gr.HTML(label="Solution (HTML with LaTeX)") with gr.Row(): openrouter_html_download = gr.File( label="📄 Download HTML Solution", visible=False ) openrouter_pdf_download = gr.File( label="📄 Download PDF Solution", visible=False ) openrouter_conversation_history = gr.State(value=None) def handle_openrouter_submit(api_key, problem_text, history): html_solution, html_file, pdf_file, updated_history = generate_math_solution_openrouter( api_key, problem_text, history ) return ( html_solution, updated_history, gr.update(value=html_file, visible=html_file is not None), gr.update(value=pdf_file, visible=pdf_file is not None) ) openrouter_submit_btn.click( fn=handle_openrouter_submit, inputs=[openrouter_api_key, text_problem_input, openrouter_conversation_history], outputs=[ openrouter_solution_output, openrouter_conversation_history, openrouter_html_download, openrouter_pdf_download ] ) def clear_openrouter(): return ( "", None, gr.update(value=None, visible=False), gr.update(value=None, visible=False) ) openrouter_clear_btn.click( fn=clear_openrouter, inputs=[], outputs=[ openrouter_solution_output, openrouter_conversation_history, openrouter_html_download, openrouter_pdf_download ] ) # 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...", 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.HTML(label="Solution (HTML with LaTeX)") with gr.Row(): together_html_download = gr.File( label="📄 Download HTML Solution", visible=False ) together_pdf_download = gr.File( label="📄 Download PDF Solution", visible=False ) together_conversation_history = gr.State(value=None) def handle_together_submit(api_key, problem_text, image_path, history): html_solution, html_file, pdf_file, updated_history = generate_math_solution_together( api_key, problem_text, image_path, history ) return ( html_solution, updated_history, gr.update(value=html_file, visible=html_file is not None), gr.update(value=pdf_file, visible=pdf_file is not None) ) together_submit_btn.click( fn=handle_together_submit, inputs=[together_api_key, together_problem_input, together_image_input, together_conversation_history], outputs=[ together_solution_output, together_conversation_history, together_html_download, together_pdf_download ] ) def clear_together(): return ( "", None, gr.update(value=None, visible=False), gr.update(value=None, visible=False) ) together_clear_btn.click( fn=clear_together, inputs=[], outputs=[ together_solution_output, together_conversation_history, together_html_download, together_pdf_download ] ) # Help tab with gr.TabItem("Help"): gr.Markdown(""" ## How to Use the Advanced Math Tutor ### Key Features 🎉 - **Professional math rendering**: All equations rendered using MathJax/LaTeX - **HTML-formatted solutions**: Beautiful, readable format - **Download HTML**: Save solutions as HTML files with proper math notation - **Download PDF**: Convert solutions to PDF (Note: PDF math rendering may vary) - **Print functionality**: Print directly from your browser ### Getting Started #### For Text Problems (OpenRouter) 1. Get API key from [OpenRouter](https://openrouter.ai/) 2. Enter your problem in proper notation 3. Get beautifully formatted solutions with proper mathematical notation #### For Image Problems (Together AI) 1. Get API key from [Together AI](https://www.together.ai/) 2. Upload a clear image of the math problem 3. Optionally add context ### Math Notation Tips - Superscripts: x² or x^2 - Subscripts: x₁ or x_1 - Fractions: Use / or describe as "numerator over denominator" - Greek letters: pi, alpha, beta, theta, etc. - Square roots: √ or sqrt() ### Types of Problems - Algebra, Calculus, Trigonometry - Geometry, Statistics - Number Theory, Linear Algebra - And much more! ### Note on PDF Export The HTML version will always have perfect math rendering. The PDF export uses WeasyPrint which may have limitations with complex math notation. For best results, use the "Print to PDF" button in the HTML view instead. """) gr.Markdown(""" --- ### About Enhanced with **MathJax** for professional mathematical notation rendering. Your API keys are never stored permanently. """) return demo # Launch the app if __name__ == "__main__": demo = create_demo() demo.launch()