Update app.py

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel, load_tool, tool
+from smolagents import CodeAgent, OpenAIServerModel, load_tool, tool
 import datetime
 import pytz
 import yaml
@@ -81,27 +81,31 @@ def initialize_agent(api_key: str):
 
     return agent
 
-# ----- Gradio Interface -----
-def launch_agent(api_key: str):
+# ----- Gradio Query Function -----
+def agent_response(api_key: str, user_query: str):
     """
-    Launches the Gradio interface for the CodeAgent after receiving the API key.
+    Responds to a user query using the CodeAgent initialized with the API key.
 
     Args:
-        api_key (str): User's OpenAI API key.
+        api_key (str): OpenAI API key provided by the user.
+        user_query (str): The user's query for the agent.
 
     Returns:
-        None
+        str: Agent's response.
     """
     agent = initialize_agent(api_key)
-    return GradioUI(agent).launch(share=True)
+    return agent.run(user_query)  # Run the agent on the user's query
 
-# Gradio input for API key
+# ----- Gradio Interface -----
 iface = gr.Interface(
-    fn=launch_agent,
-    inputs=gr.Textbox(label="Enter your OpenAI API Key", type="password"),
-    outputs="text",
+    fn=agent_response,
+    inputs=[
+        gr.Textbox(label="Enter your OpenAI API Key", type="password"),
+        gr.Textbox(label="Your Query", placeholder="Type your question here...")
+    ],
+    outputs=gr.Textbox(label="Agent Response"),
     title="Custom SmolAgent App",
-    description="Enter your OpenAI API key to use the agent."
+    description="Enter your OpenAI API key and ask a question to the agent."
 )
 
 iface.launch()