Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 import re
 import json
 import gradio as gr
-from typing import List, Dict, Any, Generator
+from typing import List, Dict, Any
 import requests
 from datetime import datetime
 import ast
@@ -275,13 +275,13 @@ def call_llm(messages: List[Dict], temperature: float = 0.7, max_tokens: int = 5
     except Exception as e:
         return f"Error calling LLM: {str(e)}"
 
-def think_only_mode(question: str) -> Generator[str, None, None]:
+def think_only_mode(question: str) -> str:
     """Think-Only mode: Chain-of-Thought only, no tools."""
     prompt = THINK_ONLY_PROMPT.format(question=question)
     messages = [{"role": "user", "content": prompt}]
     
-    yield "**Mode: Think-Only (Chain-of-Thought)**\n\n"
-    yield "Generating thoughts...\n\n"
+    output = "**Mode: Think-Only (Chain-of-Thought)**\n\n"
+    output += "Generating thoughts...\n\n"
     
     response = call_llm(messages, temperature=0.7, max_tokens=800)
     
@@ -289,20 +289,21 @@ def think_only_mode(question: str) -> Generator[str, None, None]:
     for line in lines:
         if line.strip():
             if line.strip().startswith('Thought:'):
-                yield f"**{line.strip()}**\n\n"
+                output += f"**{line.strip()}**\n\n"
             elif line.strip().startswith('Answer:'):
-                yield f"**{line.strip()}**\n\n"
+                output += f"**{line.strip()}**\n\n"
             else:
-                yield f"{line}\n\n"
+                output += f"{line}\n\n"
     
-    yield "\n---\n**Mode completed**\n"
+    output += "\n---\n**Mode completed**\n"
+    return output
 
-def act_only_mode(question: str, max_iterations: int = 5) -> Generator[str, None, None]:
+def act_only_mode(question: str, max_iterations: int = 5) -> str:
     """Act-Only mode: Tool use only, no explicit thinking."""
     tool_descriptions = get_tool_descriptions()
     prompt = ACT_ONLY_PROMPT.format(question=question, tools=tool_descriptions)
     
-    yield "**Mode: Act-Only (Tool Use Only)**\n\n"
+    output = "**Mode: Act-Only (Tool Use Only)**\n\n"
     
     messages = [{"role": "user", "content": prompt}]
     iteration = 0
@@ -315,35 +316,36 @@ def act_only_mode(question: str, max_iterations: int = 5) -> Generator[str, None
         if 'Answer:' in response:
             answer_match = re.search(r'Answer:\s*(.+)', response, re.IGNORECASE | re.DOTALL)
             if answer_match:
-                yield f"**Answer:** {answer_match.group(1).strip()}\n\n"
+                output += f"**Answer:** {answer_match.group(1).strip()}\n\n"
                 break
         
         action_name, action_input = parse_action(response)
         
         if action_name and action_input:
-            yield f"**Action:** {action_name}\n"
-            yield f"**Action Input:** {action_input}\n\n"
+            output += f"**Action:** {action_name}\n"
+            output += f"**Action Input:** {action_input}\n\n"
             
             observation = call_tool(action_name, action_input)
-            yield f"**Observation:** {observation}\n\n"
+            output += f"**Observation:** {observation}\n\n"
             
             messages.append({"role": "assistant", "content": response})
             messages.append({"role": "user", "content": f"Observation: {observation}\n\nContinue with another action or provide the final answer."})
         else:
-            yield f"Could not parse action from response. Response: {response}\n\n"
+            output += f"Could not parse action from response. Response: {response}\n\n"
             break
     
     if iteration >= max_iterations:
-        yield "**Reached maximum iterations.**\n\n"
+        output += "**Reached maximum iterations.**\n\n"
     
-    yield "\n---\n**Mode completed**\n"
+    output += "\n---\n**Mode completed**\n"
+    return output
 
-def react_mode(question: str, max_iterations: int = 5) -> Generator[str, None, None]:
+def react_mode(question: str, max_iterations: int = 5) -> str:
     """ReAct mode: Interleaving Thought, Action, Observation."""
     tool_descriptions = get_tool_descriptions()
     prompt = REACT_PROMPT.format(question=question, tools=tool_descriptions)
     
-    yield "**Mode: ReAct (Thought + Action + Observation)**\n\n"
+    output = "**Mode: ReAct (Thought + Action + Observation)**\n\n"
     
     messages = [{"role": "user", "content": prompt}]
     iteration = 0
@@ -355,34 +357,35 @@ def react_mode(question: str, max_iterations: int = 5) -> Generator[str, None, N
         
         thought_matches = re.findall(r'Thought:\s*(.+?)(?=\n(?:Action:|Answer:|$))', response, re.IGNORECASE | re.DOTALL)
         for thought in thought_matches:
-            yield f"**Thought:** {thought.strip()}\n\n"
+            output += f"**Thought:** {thought.strip()}\n\n"
         
         if 'Answer:' in response:
             answer_match = re.search(r'Answer:\s*(.+)', response, re.IGNORECASE | re.DOTALL)
             if answer_match:
-                yield f"**Answer:** {answer_match.group(1).strip()}\n\n"
+                output += f"**Answer:** {answer_match.group(1).strip()}\n\n"
                 break
         
         action_name, action_input = parse_action(response)
         
         if action_name and action_input:
-            yield f"**Action:** {action_name}\n"
-            yield f"**Action Input:** {action_input}\n\n"
+            output += f"**Action:** {action_name}\n"
+            output += f"**Action Input:** {action_input}\n\n"
             
             observation = call_tool(action_name, action_input)
-            yield f"**Observation:** {observation}\n\n"
+            output += f"**Observation:** {observation}\n\n"
             
             messages.append({"role": "assistant", "content": response})
             messages.append({"role": "user", "content": f"Observation: {observation}\n\nThought:"})
         else:
             if 'Answer:' not in response:
-                yield f"No action found. Response: {response}\n\n"
+                output += f"No action found. Response: {response}\n\n"
             break
     
     if iteration >= max_iterations:
-        yield "**Reached maximum iterations.**\n\n"
+        output += "**Reached maximum iterations.**\n\n"
     
-    yield "\n---\n**Mode completed**\n"
+    output += "\n---\n**Mode completed**\n"
+    return output
 
 EXAMPLES = [
     "What is the capital of France and what's the current weather there?",
@@ -406,6 +409,37 @@ def run_comparison(question: str, mode: str):
     else:
         return "Invalid mode selected.", "", ""
 
+def download_results(think_output: str, act_output: str, react_output: str, question: str):
+    """Create a downloadable text file with all results."""
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    filename = f"reasoning_comparison_{timestamp}.txt"
+    
+    content = f"""LLM Reasoning Modes Comparison Results
+Generated: {datetime.now().strftime("%Y-%m-%d %H:%M:%S")}
+Question: {question}
+
+{'='*80}
+THINK-ONLY MODE
+{'='*80}
+{think_output if think_output else "Not executed"}
+
+{'='*80}
+ACT-ONLY MODE
+{'='*80}
+{act_output if act_output else "Not executed"}
+
+{'='*80}
+REACT MODE
+{'='*80}
+{react_output if react_output else "Not executed"}
+"""
+    
+    filepath = f"/tmp/{filename}"
+    with open(filepath, 'w', encoding='utf-8') as f:
+        f.write(content)
+    
+    return filepath
+
 with gr.Blocks(title="LLM Reasoning Modes Comparison") as demo:
     gr.Markdown("""
     # LLM Reasoning Modes Comparison
@@ -431,7 +465,9 @@ with gr.Blocks(title="LLM Reasoning Modes Comparison") as demo:
                 value="All (Compare)",
                 label="Select Mode"
             )
-            submit_btn = gr.Button("Run", variant="primary", size="lg")
+            with gr.Row():
+                submit_btn = gr.Button("Run", variant="primary", size="lg")
+                download_btn = gr.Button("Download Results", variant="secondary", size="lg")
         
         with gr.Column(scale=1):
             gr.Markdown("**Example Questions**")
@@ -451,11 +487,22 @@ with gr.Blocks(title="LLM Reasoning Modes Comparison") as demo:
         with gr.Column():
             react_output = gr.Markdown(label="ReAct Output")
     
+    download_file = gr.File(label="Download", visible=False)
+    
     submit_btn.click(
         fn=run_comparison,
         inputs=[question_input, mode_dropdown],
         outputs=[think_output, act_output, react_output]
     )
+    
+    download_btn.click(
+        fn=download_results,
+        inputs=[think_output, act_output, react_output, question_input],
+        outputs=download_file
+    ).then(
+        fn=lambda: gr.File(visible=True),
+        outputs=download_file
+    )
 
 if __name__ == "__main__":
     demo.launch()