Update app.py

app.py

@@ -224,83 +224,35 @@ def get_festivals_today(date: str = None) -> str:
 
 final_answer = FinalAnswerTool()
 
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-with open("prompts.yaml", 'r') as stream:
-    prompt_templates = yaml.safe_load(stream)
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
 # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
 
-
-# Define Models
-primary_model = HfApiModel(
-    max_tokens=2096,
-    temperature=0.5,
-    model_id='sarvamai/sarvam-m',  # Primary model, possibly overloaded
 )
-
-backup_model = HfApiModel(
-    max_tokens=2096,
-    temperature=0.5,
-    model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',  # Backup model
+model = HfApiModel(
+max_tokens=2096,
+temperature=0.5,
+model_id='sarvamai/sarvam-m',# it is possible that this model may be overloaded
 )
 
-# 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=tools,
-        max_steps=6,
-        verbosity_level=1,
-        grammar=None,
-        planning_interval=None,
-        name=None,
-        description=None,
-        prompt_templates=prompt_templates
-    )
-
-# Instantiate Primary Agent
-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:
-        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 with error: {e}")
-        print("Switching to the backup model...")
-        try:
-            backup_agent = build_agent(backup_model)
-            result = backup_agent.run(user_input)
-            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:
-            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."
+
+# Import tool from Hub
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+
+with open("prompts.yaml", 'r') as stream:
+    prompt_templates = yaml.safe_load(stream)
+    
+agent = CodeAgent(
+    model=model,
+    tools=[final_answer], ## add your tools here (don't remove final answer)
+    max_steps=6,
+    verbosity_level=1,
+    grammar=None,
+    planning_interval=None,
+    name=None,
+    description=None,
+    prompt_templates=prompt_templates
 )
 
-iface.launch()
+
+GradioUI(agent).launch()