Update app.py

app.py

@@ -148,32 +148,35 @@ def safe_calculator(expression: str) -> str:
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+# --- Agent Definition using HfAgent ---
 class HfAgentWrapper:
     def __init__(self):
         print("Initializing HfAgentWrapper...")
-        model_repo = "google/gemma-1.1-7b-it"  # <<< --- CORRECTED MODEL REPO ID
+        model_repo = "google/gemma-1.1-7b-it"  # Correct Hugging Face Hub ID
         print(f"Loading HfAgent with model: {model_repo}")
 
-    try:
-        if not get_current_time_in_timezone.__doc__: 
-            raise ValueError("Tool 'get_current_time_in_timezone' is missing a docstring.")
-        if not web_search.__doc__: 
-            raise ValueError("Tool 'web_search' is missing a docstring.")
-        if not safe_calculator.__doc__: 
-            raise ValueError("Tool 'safe_calculator' is missing a docstring.")
-
+        try:
+            # Ensure your tool functions have docstrings.
+            if not get_current_time_in_timezone.__doc__: 
+                raise ValueError("Tool 'get_current_time_in_timezone' is missing a docstring.")
+            if not web_search.__doc__: 
+                raise ValueError("Tool 'web_search' is missing a docstring.")
+            if not safe_calculator.__doc__: 
+                raise ValueError("Tool 'safe_calculator' is missing a docstring.")
+
+            # Manually wrap your functions into Tool objects
             time_tool_obj = Tool(
-                name="get_current_time_in_timezone",
+                name=get_current_time_in_timezone.__name__,
                 func=get_current_time_in_timezone,
                 description=get_current_time_in_timezone.__doc__
             )
             search_tool_obj = Tool(
-                name="web_search",
+                name=web_search.__name__,
                 func=web_search,
                 description=web_search.__doc__
             )
             calculator_tool_obj = Tool(
-                name="safe_calculator",  # <--- EXPLICIT NAME
+                name=safe_calculator.__name__,
                 func=safe_calculator,
                 description=safe_calculator.__doc__
             )
@@ -183,22 +186,21 @@ class HfAgentWrapper:
             tool_names_for_log = [tool.name for tool in self.actual_tools_for_agent]
             print(f"Prepared Tool objects with names: {tool_names_for_log} and descriptions.")
 
-            # Initialize HfAgent with the corrected model_repo and without the 'tools=[]' argument
+            # Initialize HfAgent, now including trust_remote_code=True
             self.agent = HfAgent(
-                model_repo, # Uses the corrected model_repo string
-                additional_tools=self.actual_tools_for_agent
+                model_repo, 
+                additional_tools=self.actual_tools_for_agent,
+                trust_remote_code=True  # <<< --- ENSURE THIS IS ADDED CORRECTLY WITH A COMMA BEFORE IT
             )
             print(f"HfAgent successfully instantiated with model {model_repo}.")
 
-            # --- Robust toolbox inspection (keep this as it is) ---
+            # --- Robust toolbox inspection ---
             print(f"Inspecting self.agent.toolbox...")
-            # ... (the rest of your toolbox inspection code from the previous version) ...
             if hasattr(self.agent, 'toolbox'):
                 toolbox_attr = self.agent.toolbox
                 print(f"Type of self.agent.toolbox: {type(toolbox_attr)}")
 
                 if isinstance(toolbox_attr, dict):
-                    # ... (dictionary inspection logic) ...
                     print(f"Toolbox is a dict. Keys: {list(toolbox_attr.keys())}")
                     tools_info_from_dict = {}
                     for key, value in list(toolbox_attr.items())[:5]: 
@@ -208,6 +210,7 @@ class HfAgentWrapper:
                         tools_info_from_dict[key] = {'name_attr': tool_name_in_dict, 'type': str(type(value))}
                     print(f"Sample of toolbox dict content: {tools_info_from_dict}")
                     print(f"Names of tools we passed to HfAgent: {[tool.name for tool in self.actual_tools_for_agent]}")
+
                 elif hasattr(toolbox_attr, 'tools') and callable(toolbox_attr.tools):
                     try:
                         tools_in_toolbox = toolbox_attr.tools()
@@ -220,14 +223,13 @@ class HfAgentWrapper:
                 print("self.agent does not have a 'toolbox' attribute (this would be unexpected).")
             # --- End of robust toolbox inspection ---
 
-            except Exception as e:
-                print(f"CRITICAL ERROR: Failed to initialize HfAgent: {e}")
-                import traceback 
-                print("Full traceback of HfAgent initialization error:")
-                traceback.print_exc() 
-                raise RuntimeError(f"HfAgent initialization failed: {e}") from e
+        except Exception as e: # This is the main except block for the __init__ try
+            print(f"CRITICAL ERROR: Failed to initialize HfAgent: {e}")
+            # import traceback # Already imported at the top of your file
+            print("Full traceback of HfAgent initialization error:")
+            traceback.print_exc() 
+            raise RuntimeError(f"HfAgent initialization failed: {e}") from e
 
-    # __call__ method remains the same
     def __call__(self, question: str) -> str:
         print(f"\n--- HfAgentWrapper received question (first 100 chars): {question[:100]}... ---")
         try:
@@ -236,7 +238,7 @@ class HfAgentWrapper:
             return str(answer)
         except Exception as e:
             print(f"ERROR: HfAgent execution failed for question '{question[:50]}...': {e}")
-            import traceback
+            # import traceback # Already imported at the top of your file
             print("Full traceback of HfAgent execution error:")
             traceback.print_exc()
             return f"Agent Error: Failed to process the question. Details: {e}"