Update app.py

app.py

@@ -9,7 +9,6 @@ import ast
 import operator as op
 import wikipedia
 
-# Tool implementations
 class Tool:
     def __init__(self, name: str, description: str, func):
         self.name = name
@@ -32,7 +31,6 @@ def duckduckgo_search(query: str) -> str:
         response = requests.get(url, params=params, timeout=10)
         data = response.json()
         
-        # Get abstract or first related topic
         if data.get('Abstract'):
             return f"Search result: {data['Abstract']}"
         elif data.get('RelatedTopics') and len(data['RelatedTopics']) > 0:
@@ -50,7 +48,6 @@ def wikipedia_search(query: str) -> str:
     """Search Wikipedia for information."""
     try:
         wikipedia.set_lang("en")
-        # Get summary
         summary = wikipedia.summary(query, sentences=3, auto_suggest=True)
         return f"Wikipedia: {summary}"
     except wikipedia.exceptions.DisambiguationError as e:
@@ -80,7 +77,6 @@ def get_weather(location: str) -> str:
 
 def calculate(expression: str) -> str:
     """Safely evaluate mathematical expressions."""
-    # Supported operators
     operators = {
         ast.Add: op.add,
         ast.Sub: op.sub,
@@ -99,7 +95,6 @@ def calculate(expression: str) -> str:
         elif isinstance(node, ast.UnaryOp):
             return operators[type(node.op)](eval_expr(node.operand))
         elif isinstance(node, ast.Call):
-            # Support basic math functions
             if node.func.id == 'abs':
                 return abs(eval_expr(node.args[0]))
             elif node.func.id == 'round':
@@ -108,9 +103,7 @@ def calculate(expression: str) -> str:
             raise TypeError(node)
     
     try:
-        # Clean the expression
         expression = expression.strip()
-        # Parse and evaluate
         node = ast.parse(expression, mode='eval')
         result = eval_expr(node.body)
         return f"Result: {result}"
@@ -120,7 +113,6 @@ def calculate(expression: str) -> str:
 def python_repl(code: str) -> str:
     """Execute safe Python code (limited to basic operations)."""
     try:
-        # Whitelist of safe builtins
         safe_builtins = {
             'abs': abs, 'round': round, 'min': min, 'max': max,
             'sum': sum, 'len': len, 'range': range, 'list': list,
@@ -129,23 +121,18 @@ def python_repl(code: str) -> str:
             'sorted': sorted, 'reversed': reversed,
         }
         
-        # Create restricted namespace
         namespace = {'__builtins__': safe_builtins}
         
-        # Capture output
         from io import StringIO
         import sys
         old_stdout = sys.stdout
         sys.stdout = StringIO()
         
-        # Execute code
         exec(code, namespace)
         
-        # Get output
         output = sys.stdout.getvalue()
         sys.stdout = old_stdout
         
-        # Also get any variables that were set
         result_vars = {k: v for k, v in namespace.items() if k != '__builtins__' and not k.startswith('_')}
         
         result = output if output else str(result_vars) if result_vars else "Code executed successfully (no output)"
@@ -153,7 +140,6 @@ def python_repl(code: str) -> str:
     except Exception as e:
         return f"Python error: {str(e)}"
 
-# Define tools
 TOOLS = [
     Tool(
         name="duckduckgo_search",
@@ -182,14 +168,12 @@ TOOLS = [
     ),
 ]
 
-# Create tool descriptions for prompt
 def get_tool_descriptions() -> str:
     descriptions = []
     for tool in TOOLS:
         descriptions.append(f"- {tool.name}: {tool.description}")
     return "\n".join(descriptions)
 
-# Agent prompts
 THINK_ONLY_PROMPT = """You are a helpful AI assistant. You solve problems by thinking through them step-by-step.
 
 For each question:
@@ -297,11 +281,10 @@ def think_only_mode(question: str) -> Generator[str, None, None]:
     messages = [{"role": "user", "content": prompt}]
     
     yield "**Mode: Think-Only (Chain-of-Thought)**\n\n"
-    yield "🤔 Generating thoughts...\n\n"
+    yield "Generating thoughts...\n\n"
     
     response = call_llm(messages, temperature=0.7, max_tokens=800)
     
-    # Parse and format the response
     lines = response.split('\n')
     for line in lines:
         if line.strip():
@@ -329,33 +312,29 @@ def act_only_mode(question: str, max_iterations: int = 5) -> Generator[str, None
         
         response = call_llm(messages, temperature=0.5, max_tokens=300)
         
-        # Check for final answer
         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"
+                yield f"**Answer:** {answer_match.group(1).strip()}\n\n"
                 break
         
-        # Parse action
         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"
+            yield f"**Action:** {action_name}\n"
+            yield f"**Action Input:** {action_input}\n\n"
             
-            # Execute tool
             observation = call_tool(action_name, action_input)
-            yield f" **Observation:** {observation}\n\n"
+            yield f"**Observation:** {observation}\n\n"
             
-            # Add to conversation
             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"
+            yield f"Could not parse action from response. Response: {response}\n\n"
             break
     
     if iteration >= max_iterations:
-        yield "⚠️ **Reached maximum iterations.**\n\n"
+        yield "**Reached maximum iterations.**\n\n"
     
     yield "\n---\n**Mode completed**\n"
 
@@ -374,44 +353,37 @@ def react_mode(question: str, max_iterations: int = 5) -> Generator[str, None, N
         
         response = call_llm(messages, temperature=0.7, max_tokens=400)
         
-        # Parse thoughts
         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"
         
-        # Check for final answer
         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"
                 break
         
-        # Parse action
         action_name, action_input = parse_action(response)
         
         if action_name and action_input:
-            yield f" **Action:** {action_name}\n"
+            yield f"**Action:** {action_name}\n"
             yield f"**Action Input:** {action_input}\n\n"
             
-            # Execute tool
             observation = call_tool(action_name, action_input)
             yield f"**Observation:** {observation}\n\n"
             
-            # Add to conversation
             messages.append({"role": "assistant", "content": response})
             messages.append({"role": "user", "content": f"Observation: {observation}\n\nThought:"})
         else:
-            # If no action but also no answer, there might be an issue
             if 'Answer:' not in response:
-                yield f"⚠️ No action found. Response: {response}\n\n"
+                yield f"No action found. Response: {response}\n\n"
             break
     
     if iteration >= max_iterations:
-        yield "⚠️ **Reached maximum iterations.**\n\n"
+        yield "**Reached maximum iterations.**\n\n"
     
     yield "\n---\n**Mode completed**\n"
 
-# Example questions
 EXAMPLES = [
     "What is the capital of France and what's the current weather there?",
     "Who wrote 'To Kill a Mockingbird' and when was it published?",
@@ -434,19 +406,17 @@ def run_comparison(question: str, mode: str):
     else:
         return "Invalid mode selected.", "", ""
 
-# Gradio Interface
 with gr.Blocks(title="LLM Reasoning Modes Comparison") as demo:
     gr.Markdown("""
-    # 🧠 LLM Reasoning Modes Comparison
+    # LLM Reasoning Modes Comparison
     
     Compare three reasoning approaches using **openai/gpt-oss-20b**:
     
     - **Think-Only**: Chain-of-Thought reasoning only (no tools)
     - **Act-Only**: Tool use only (no explicit reasoning)
-    - **ReAct**: Interleaved Thought → Action → Observation
+    - **ReAct**: Interleaved Thought, Action, Observation
     
-    ### Available Tools:
-    🔍 DuckDuckGo Search | 📚 Wikipedia | 🌤️ Weather API | 🧮 Calculator | 🐍 Python REPL
+    Available Tools: DuckDuckGo Search | Wikipedia | Weather API | Calculator | Python REPL
     """)
     
     with gr.Row():
@@ -461,12 +431,12 @@ 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")
+            submit_btn = gr.Button("Run", variant="primary", size="lg")
         
         with gr.Column(scale=1):
-            gr.Markdown("###  Example Questions")
+            gr.Markdown("**Example Questions**")
             for idx, example in enumerate(EXAMPLES):
-                gr.Button(f"Ex {idx+1}", size="sm").click(
+                gr.Button(f"Example {idx+1}", size="sm").click(
                     fn=lambda ex=example: ex,
                     outputs=question_input
                 )
@@ -486,7 +456,6 @@ with gr.Blocks(title="LLM Reasoning Modes Comparison") as demo:
         inputs=[question_input, mode_dropdown],
         outputs=[think_output, act_output, react_output]
     )
-    
-  
+
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()