File size: 28,517 Bytes
fbe34cb
 
 
 
871e4ec
ca5e34e
 
871e4ec
 
fd80886
 
 
 
fbe34cb
fd80886
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ca5e34e
fbe34cb
fd80886
fbe34cb
 
fd80886
fbe34cb
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
fd80886
fbe34cb
fd80886
 
 
 
 
 
 
 
fbe34cb
 
 
 
 
 
fd80886
fbe34cb
 
 
 
 
 
 
 
 
 
 
 
 
 
 
fd80886
 
 
 
 
 
 
 
 
 
fbe34cb
 
 
 
fd80886
fbe34cb
fd80886
fbe34cb
 
 
fd80886
871e4ec
fd80886
871e4ec
ca5e34e
fd80886
ca5e34e
871e4ec
fd80886
ca5e34e
 
 
 
871e4ec
 
 
 
 
 
 
 
 
 
 
fd80886
871e4ec
fd80886
 
 
 
 
 
 
 
871e4ec
 
ca5e34e
 
 
 
 
fd80886
ca5e34e
 
871e4ec
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ca5e34e
 
 
 
 
 
 
 
fd80886
 
 
 
 
 
 
 
 
 
 
ca5e34e
 
 
 
fd80886
ca5e34e
fd80886
ca5e34e
 
 
fd80886
 
 
 
 
 
 
 
 
 
 
 
 
ca5e34e
fbe34cb
 
 
 
 
ca5e34e
fd80886
ca5e34e
 
fbe34cb
 
 
 
ca5e34e
 
fbe34cb
 
ca5e34e
fbe34cb
 
 
 
ca5e34e
fbe34cb
 
 
 
85415db
 
 
 
 
 
 
ca5e34e
85415db
 
fbe34cb
ca5e34e
 
fd80886
fbe34cb
fd80886
 
 
 
 
 
 
 
 
 
 
fbe34cb
 
ca5e34e
fbe34cb
 
fd80886
 
 
 
 
 
 
 
 
 
 
 
 
ca5e34e
fd80886
ca5e34e
fd80886
 
 
 
 
 
fbe34cb
 
fd80886
 
 
 
 
 
 
 
ca5e34e
fd80886
fbe34cb
fd80886
 
 
 
 
 
fbe34cb
 
ca5e34e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
fd80886
ca5e34e
fd80886
 
 
 
 
 
 
 
 
 
 
ca5e34e
 
 
 
 
fd80886
 
 
 
 
 
 
 
 
 
 
 
 
ca5e34e
fd80886
ca5e34e
fd80886
 
 
 
 
 
ca5e34e
 
fd80886
 
 
 
 
 
 
 
ca5e34e
fd80886
ca5e34e
fd80886
 
 
 
 
 
ca5e34e
 
 
fbe34cb
 
 
 
fd80886
 
 
 
 
 
fbe34cb
ca5e34e
 
 
fbe34cb
ca5e34e
 
 
 
 
 
 
 
fbe34cb
 
ca5e34e
 
fd80886
 
 
 
 
 
 
 
 
 
fbe34cb
 
 
 
 
85415db
 
ca5e34e
fbe34cb
85415db
 
 
 
 
 
 
 
fd80886
 
 
 
 
 
fbe34cb
 
 
 
 
 
fd80886
ca5e34e
fd80886
 
 
 
 
 
 
 
ca5e34e
fbe34cb
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
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
from datetime import datetime
import tempfile
import weasyprint
from pathlib import Path

# Function to convert markdown to HTML with styling
def markdown_to_html(markdown_text, problem_text="", include_problem=True):
    """Convert markdown to styled HTML"""
    
    # Convert markdown to HTML
    html_content = markdown.markdown(markdown_text, extensions=['tables', 'fenced_code'])
    
    # Get current timestamp
    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
    
    # Create styled HTML document
    styled_html = f"""
    <!DOCTYPE html>
    <html lang="en">
    <head>
        <meta charset="UTF-8">
        <meta name="viewport" content="width=device-width, initial-scale=1.0">
        <title>Math Solution - Advanced Math Tutor</title>
        <style>
            body {{
                font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
                line-height: 1.6;
                color: #333;
                max-width: 800px;
                margin: 0 auto;
                padding: 20px;
                background-color: #f9f9f9;
            }}
            .container {{
                background-color: white;
                padding: 40px;
                border-radius: 10px;
                box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
            }}
            .header {{
                text-align: center;
                border-bottom: 3px solid #4CAF50;
                padding-bottom: 20px;
                margin-bottom: 30px;
            }}
            .header h1 {{
                color: #2c3e50;
                margin: 0;
                font-size: 2.5em;
            }}
            .header .subtitle {{
                color: #7f8c8d;
                font-style: italic;
                margin-top: 10px;
            }}
            .problem-section {{
                background-color: #e8f5e8;
                padding: 20px;
                border-radius: 8px;
                margin-bottom: 30px;
                border-left: 5px solid #4CAF50;
            }}
            .problem-section h2 {{
                color: #2c3e50;
                margin-top: 0;
            }}
            .solution-content {{
                background-color: #f8f9fa;
                padding: 25px;
                border-radius: 8px;
                border-left: 5px solid #007bff;
            }}
            h1, h2, h3, h4, h5, h6 {{
                color: #2c3e50;
                margin-top: 25px;
                margin-bottom: 15px;
            }}
            h2 {{
                border-bottom: 2px solid #eee;
                padding-bottom: 10px;
            }}
            code {{
                background-color: #f1f1f1;
                padding: 2px 6px;
                border-radius: 3px;
                font-family: 'Courier New', monospace;
                color: #d63384;
            }}
            pre {{
                background-color: #f8f8f8;
                padding: 15px;
                border-radius: 5px;
                overflow-x: auto;
                border: 1px solid #ddd;
            }}
            pre code {{
                background-color: transparent;
                padding: 0;
                color: inherit;
            }}
            .math-expression {{
                background-color: #fff3cd;
                padding: 10px;
                border-radius: 5px;
                border: 1px solid #ffeaa7;
                margin: 10px 0;
                font-family: 'Times New Roman', serif;
                font-size: 1.1em;
            }}
            .step {{
                margin: 20px 0;
                padding: 15px;
                background-color: #ffffff;
                border-radius: 8px;
                border: 1px solid #dee2e6;
            }}
            .final-answer {{
                background-color: #d4edda;
                border: 2px solid #4CAF50;
                padding: 20px;
                border-radius: 8px;
                margin-top: 30px;
                text-align: center;
                font-weight: bold;
                font-size: 1.2em;
            }}
            .timestamp {{
                text-align: right;
                color: #6c757d;
                font-size: 0.9em;
                margin-top: 30px;
                padding-top: 20px;
                border-top: 1px solid #eee;
            }}
            ul, ol {{
                padding-left: 25px;
            }}
            li {{
                margin: 8px 0;
            }}
            table {{
                border-collapse: collapse;
                width: 100%;
                margin: 20px 0;
            }}
            th, td {{
                border: 1px solid #ddd;
                padding: 12px;
                text-align: left;
            }}
            th {{
                background-color: #f8f9fa;
                font-weight: bold;
            }}
            .print-button {{
                background-color: #007bff;
                color: white;
                border: none;
                padding: 10px 20px;
                border-radius: 5px;
                cursor: pointer;
                font-size: 16px;
                margin: 10px 5px;
                display: inline-block;
                text-decoration: none;
            }}
            .print-button:hover {{
                background-color: #0056b3;
            }}
            @media print {{
                body {{
                    background-color: white;
                }}
                .container {{
                    box-shadow: none;
                    border: none;
                }}
                .print-button {{
                    display: none;
                }}
            }}
        </style>
        <script>
            function printPage() {{
                window.print();
            }}
        </script>
    </head>
    <body>
        <div class="container">
            <div class="header">
                <h1>πŸ“š Advanced Math Tutor</h1>
                <div class="subtitle">Step-by-Step Mathematical Solution</div>
            </div>
            
            <button class="print-button" onclick="printPage()">πŸ–¨οΈ Print to PDF</button>
            
            {f'''
            <div class="problem-section">
                <h2>πŸ“ Problem Statement</h2>
                <p><strong>{problem_text}</strong></p>
            </div>
            ''' if include_problem and problem_text.strip() else ''}
            
            <div class="solution-content">
                <h2>πŸ” Solution</h2>
                {html_content}
            </div>
            
            <div class="timestamp">
                Generated on: {timestamp}
            </div>
        </div>
    </body>
    </html>
    """
    
    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
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:
        # Convert HTML to PDF using WeasyPrint
        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
             2. Explain why this step is necessary
             3. Connect it to relevant mathematical concepts
             
             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
             - `Code blocks` for mathematical expressions
             - 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]:  # Check if there's a response
                    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="microsoft/phi-4-reasoning-plus: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
                 2. Explain why this step is necessary
                 3. Connect it to relevant mathematical concepts
                 
                 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
                 - `Code blocks` for mathematical expressions
                 - 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]:  # Check if there's a response
                    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="meta-llama/Llama-Vision-Free",
            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 - for simplicity, just store the text problem
        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 download and print-to-PDF capabilities.
        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():
            # 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 the equation: 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 Format)")
                    
                    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
                        )
                
                # Store conversation history (invisible to user)
                openrouter_conversation_history = gr.State(value=None)
                
                # Button actions
                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 outputs including file updates
                    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 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.HTML(label="Solution (HTML Format)")
                    
                    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
                        )
                
                # Store conversation history (invisible to user)
                together_conversation_history = gr.State(value=None)
                
                # Button actions
                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 outputs including file updates
                    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
                
                ### New Features πŸŽ‰
                - **HTML-formatted solutions**: All responses are now generated in beautiful HTML format
                - **Download HTML**: Download the complete solution as an HTML file
                - **Download PDF**: Convert and download solutions as PDF files
                - **Print functionality**: Use the "Print to PDF" button in the HTML output to print directly
                
                ### Getting Started
                
                #### 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 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
                - 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 in HTML format
                - Use the download buttons to save HTML or PDF versions
                - Click "Print to PDF" within the solution to print directly from your browser
                
                ### HTML Output Features
                - **Professional styling**: Clean, readable format with proper typography
                - **Mathematical expressions**: Highlighted math expressions and code blocks
                - **Step-by-step sections**: Clearly organized solution steps
                - **Print-friendly**: Optimized for printing and PDF conversion
                - **Timestamps**: Each solution includes generation timestamp
                
                ### Tips for Best Results
                - Be specific in your problem description
                - Include all necessary information
                - 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)
                - Calculus (derivatives, integrals, limits)
                - Trigonometry
                - Geometry
                - Statistics and Probability
                - Number Theory
                - And many more!
                
                ### Required Dependencies
                To run this application, you'll need to install:
                ```bash
                pip install gradio openai together pillow markdown weasyprint
                ```
                """)
        
        # Footer
        gr.Markdown("""
        ---
        ### About
        This enhanced 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. 
        
        **New Features:**
        - HTML-formatted responses with professional styling
        - Download solutions as HTML files
        - Convert and download solutions as PDF files
        - Print-to-PDF functionality
        - Enhanced formatting with mathematical expressions highlighting
        
        Your API keys are required but not stored permanently.
        """)
    
    return demo

# Launch the app
if __name__ == "__main__":
    demo = create_demo()
    demo.launch()