Update app.py

app.py

@@ -45,18 +45,17 @@ model = HfApiModel(
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
-def run_agent(user_input):
+def run_agent(user_input, stream=False, reset=False, additional_args=None):
     """
     Constructs a prompt from the system prompt and the user input,
     then directly calls the model to generate an answer.
     After generation, it removes any chain-of-thought, debug logs, or code snippets.
+    Accepts extra keyword arguments to be compatible with Gradio's expectations.
     """
     prompt = f"{system_prompt}\nUser Input: {user_input}\nFinal Answer:"
     try:
-        # Try using the model's run method if available.
         raw_response = model.run(prompt)
     except AttributeError:
-        # Otherwise, assume the model is callable.
         raw_response = model(prompt)
     
     print("Raw Model Response:", raw_response)  # Debug output (optional)
@@ -70,12 +69,13 @@ def run_agent(user_input):
         return "I'm unable to generate a meaningful response. Please refine your query."
     return clean_response
 
-# Create a simple agent wrapper with a run method.
 class SimpleAgent:
     def __init__(self, run_function):
-        self.run = run_function
+        self.run_function = run_function
+
+    def run(self, *args, **kwargs):
+        return self.run_function(*args, **kwargs)
 
 agent = SimpleAgent(run_agent)
 
-# Launch the Gradio UI using the simple agent.
 GradioUI(agent).launch()