Update app.py

app.py

@@ -7,6 +7,7 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
+import gradio as gr
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 
@@ -246,19 +247,15 @@ backup_model = HfApiModel(
 
 # Function to Build an Agent with Any Model
 def build_agent(model_to_use):
+    # Assuming the tools are defined elsewhere
+    tools = [
+        # final_answer, get_horoscope, get_date_panchang,
+        # get_holidays, get_panchang_field, get_festivals_today,
+        # get_current_time_in_timezone, my_custom_tool, image_generation_tool
+    ]
     return CodeAgent(
         model=model_to_use,
-        tools=[
-            final_answer,
-            get_horoscope,
-            get_date_panchang,
-            get_holidays,
-            get_panchang_field,
-            get_festivals_today,
-            get_current_time_in_timezone,
-            my_custom_tool,
-            image_generation_tool
-        ],
+        tools=tools,
         max_steps=6,
         verbosity_level=1,
         grammar=None,
@@ -269,27 +266,41 @@ def build_agent(model_to_use):
     )
 
 # Instantiate Primary Agent
-agent = build_agent(primary_model)
+primary_agent = build_agent(primary_model)
 
 # Fallback-Handled Runner Function
 def agent_runner(user_input):
+    """
+    This function takes user input, tries the primary agent,
+    and switches to the backup agent on failure.
+    """
     try:
-        result = agent.run(user_input)
-        if result is None or result.strip() == "":
-            raise ValueError("Primary model returned empty response.")
+        print("Attempting to run with the primary model...")
+        result = primary_agent.run(user_input)
+        if result is None or (isinstance(result, str) and result.strip() == ""):
+            raise ValueError("Primary model returned an empty or null response.")
         return result
     except Exception as e:
-        print(f"Primary model failed: {e}")
-        print("Switching to backup model...")
-
-        backup_agent = build_agent(backup_model)
+        print(f"Primary model failed with error: {e}")
+        print("Switching to the backup model...")
         try:
+            backup_agent = build_agent(backup_model)
             result = backup_agent.run(user_input)
-            return result or "Backup model also failed."
+            if result is None or (isinstance(result, str) and result.strip() == ""):
+                return "Backup model also returned an empty response."
+            return result
         except Exception as e2:
-            return f"Backup model failed as well: {e2}"
-
-# Launch Gradio with Fallback Logic
-
+            print(f"Backup model also failed with error: {e2}")
+            return f"The backup model failed to generate a response: {e2}"
+
+# Launch Gradio with the runner function
+# We use the standard gr.Interface for this logic.
+iface = gr.Interface(
+    fn=agent_runner,
+    inputs="text",
+    outputs="text",
+    title="Agent with Fallback Logic",
+    description="Enter your query. The system will use a primary model and switch to a backup if the primary fails."
+)
 
-GradioUI(agent_runner).launch()
+iface.launch()