Update app.py

app.py

@@ -305,8 +305,8 @@ def generate_math_solution_openrouter(api_key, problem_text, history=None):
         # 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(
+        # Create the completion with streaming
+        stream = client.chat.completions.create(
             model="deepseek/deepseek-r1-0528:free",
             messages=messages,
             stream=True,
@@ -316,7 +316,13 @@ def generate_math_solution_openrouter(api_key, problem_text, history=None):
             }
         )
         
-        markdown_solution = completion.choices[0].message.content
+        # Collect streamed content
+        markdown_solution = ""
+        for chunk in stream:
+            if hasattr(chunk, 'choices') and chunk.choices:
+                delta = chunk.choices[0].delta
+                if hasattr(delta, 'content') and delta.content:
+                    markdown_solution += delta.content
         
         # Convert to HTML
         html_solution = markdown_to_html(markdown_solution, problem_text)
@@ -413,14 +419,20 @@ def generate_math_solution_together(api_key, problem_text, image_path=None, hist
             "content": user_message_content
         })
         
-        # Create the completion
-        response = client.chat.completions.create(
+        # Create the completion with streaming
+        stream = client.chat.completions.create(
             model="meta-llama/Llama-Vision-Free",
             messages=messages,
             stream=True
         )
         
-        markdown_solution = response.choices[0].message.content
+        # Collect streamed content
+        markdown_solution = ""
+        for chunk in stream:
+            if hasattr(chunk, 'choices') and chunk.choices:
+                delta = chunk.choices[0].delta
+                if hasattr(delta, 'content') and delta.content:
+                    markdown_solution += delta.content
         
         # Convert to HTML
         problem_display = problem_text if problem_text.strip() else "Image-based problem"
@@ -432,7 +444,7 @@ def generate_math_solution_together(api_key, problem_text, image_path=None, hist
         # Convert to PDF
         pdf_file_path = html_to_pdf(html_solution, "together_solution")
         
-        # Update history - for simplicity, just store the text problem
+        # Update history
         if history is None:
             history = []
         history.append((problem_display, markdown_solution))