Update app.py

app.py

@@ -4,11 +4,23 @@ import gradio as gr
 from groq import Groq
 import re
 
-# Initialize Groq client
+# Initialize Groq
 client = Groq(api_key=os.getenv("GROQ_API_KEY"))
-
 MODEL = "llama-3.3-70b-versatile"
 
+# ------------------ Direct Answer ------------------
+def answer_anything(question):
+    response = client.chat.completions.create(
+        model=MODEL,
+        messages=[
+            {"role": "system", "content": "You are a helpful assistant."},
+            {"role": "user", "content": question}
+        ],
+        temperature=0.5
+    )
+    return response.choices[0].message.content
+
+# ------------------ Autonomous Agent ------------------
 class AutonomousAgent:
     def __init__(self, goal):
         self.goal = goal
@@ -29,7 +41,6 @@ class AutonomousAgent:
     def plan_tasks(self):
         prompt = f"""
         Goal: {self.goal}
-
         Create a step-by-step task plan to achieve this goal.
         Use short, clear tasks.
         """
@@ -48,7 +59,7 @@ class AutonomousAgent:
         """
         return self.think(prompt)
 
-    def run(self, max_steps=6):
+    def run(self, max_steps=6, show_steps=False):
         tasks_text = self.plan_tasks()
         tasks = [re.sub(r'^\d+\.\s*', '', t).strip() for t in tasks_text.split("\n") if t.strip()]
 
@@ -57,21 +68,34 @@ class AutonomousAgent:
             result = self.execute_task(task)
             self.memory.append({"task": task, "result": result})
             self.completed_tasks.append(task)
-            results.append(f"STEP {i}: {task}\nRESULT: {result}\n{'-'*40}")
-            time.sleep(1)
-        return "\n".join(results)
+            if show_steps:
+                results.append(f"STEP {i}: {task}\nRESULT: {result}\n{'-'*40}")
+
+        # Return either detailed steps or just the final result
+        if show_steps:
+            return "\n".join(results)
+        else:
+            return result  # last task result only
 
-# Gradio interface
-def run_agent(goal, max_steps=6):
-    agent = AutonomousAgent(goal)
-    return agent.run(max_steps=int(max_steps))
+# ------------------ Gradio Interface ------------------
+def run(user_input, mode, max_steps, show_steps):
+    if mode == "Direct Answer":
+        return answer_anything(user_input)
+    elif mode == "Autonomous Agent":
+        agent = AutonomousAgent(user_input)
+        return agent.run(max_steps=max_steps, show_steps=show_steps)
 
 interface = gr.Interface(
-    fn=run_agent,
-    inputs=[gr.Textbox(label="Enter Goal"), gr.Slider(1, 10, value=6, step=1, label="Max Steps")],
+    fn=run,
+    inputs=[
+        gr.Textbox(label="Enter your question or goal"),
+        gr.Radio(choices=["Direct Answer", "Autonomous Agent"], label="Mode"),
+        gr.Slider(1, 10, value=6, step=1, label="Max Steps (for Autonomous Agent)"),
+        gr.Checkbox(label="Show steps (for Autonomous Agent)", value=True)
+    ],
     outputs="text",
-    title="🤖 Autonomous AI Agent",
-    description="Enter a goal, and the agent will plan and execute tasks to achieve it."
+    title="🤖 Combined AI Agent",
+    description="Choose 'Direct Answer' for quick responses, or 'Autonomous Agent' for multi-step goal execution."
 )
 
 if __name__ == "__main__":