Update app.py

app.py

@@ -78,66 +78,19 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {e}"
 
-# -----------------------------------------------------------------------------
-# 1b) FILE-UTILITIES
-# -----------------------------------------------------------------------------
-@tool
-def list_uploaded_files() -> str:
-    """
-    Lists all files and directories in /mnt/data (where uploads appear).
-    """
-    try:
-        items = os.listdir("/mnt/data")
-        if not items:
-            return "No files uploaded."
-        return ", ".join(items)
-    except Exception as e:
-        return f"Error listing files: {e}"
-
-@tool
-def read_file(path: str) -> str:
-    """
-    Reads a text file from /mnt/data and returns its contents.
-
-    Args:
-        path: Filename under /mnt/data to read.
-    """
-    full = os.path.join("/mnt/data", path)
-    try:
-        with open(full, "r") as f:
-            return f.read()
-    except Exception as e:
-        return f"Error reading {path}: {e}"
-
-@tool
-def read_excel(path: str) -> str:
-    """
-    Reads an Excel file from /mnt/data and returns its contents as a markdown table.
-
-    Args:
-        path: Filename under /mnt/data to read.
-    """
-    full = os.path.join("/mnt/data", path)
-    try:
-        df = pd.read_excel(full)
-        # pandas.DataFrame.to_markdown requires tabulate
-        return df.to_markdown(index=False)
-    except Exception as e:
-        return f"Error reading Excel {path}: {e}"
-
 # -----------------------------------------------------------------------------
 # 2) LOAD PROMPTS & RAG MAP
 # -----------------------------------------------------------------------------
 with open("prompts.yaml", "r") as f:
     prompt_templates = yaml.safe_load(f)
 
-# extract the fallback map from YAML
+# **make sure your prompts.yaml has a top-level key `rag_fallbacks:`**
 rag_fallbacks = prompt_templates.get("rag_fallbacks", {})
 
 @tool
 def rag_fallback(question: str) -> str:
     """
-    Returns a pre‐cached answer if the question matches any RAG key.
+    Returns a pre-cached answer if the question matches any RAG key.
 
     Args:
         question: The user’s question.
@@ -171,12 +124,9 @@ class BasicAgent:
                 get_weather,
                 generate_pun,
                 web_search,
-                list_uploaded_files,
-                read_file,
-                read_excel,
-                rag_fallback,            # fallback now a tool
+                rag_fallback,        # ← our new fallback tool
             ],
-            max_steps=8,               # extra steps for file inspection + fallback
+            max_steps=7,             # ← allow one extra step for fallback
             verbosity_level=1,
             name="Son_of_Anton",
             description="Your friendly AI agent to help you with any task",
@@ -185,23 +135,25 @@ class BasicAgent:
         self.rag_fallbacks = rag_fallbacks
 
     def __call__(self, question: str) -> str:
-        # 1) run the agent through all steps (including tool-based fallback)
+        # 1) run the agent (it now has rag_fallback available as step 7)
         try:
             raw_ans = self.agent.run(question)
         except Exception:
             raw_ans = ""
-        ans_str = str(raw_ans)
 
-        # 2) if still nothing / apology, do manual fallback lookup
-        is_empty      = not ans_str.strip()
-        is_sorry      = ans_str.lower().startswith("i'm sorry")
-        is_unanswered = any(marker in ans_str.lower() for marker in ["unanswered", "failed", "none"])
-        if is_empty or is_sorry or is_unanswered:
-            q_low = question.lower()
+        ans_str = str(raw_ans).strip().lower()
+        # 2) manual fallback if still empty/apologetic/unhelpful
+        if (
+            not ans_str
+            or ans_str.startswith("i'm sorry")
+            or "unanswered"   in ans_str
+            or "none"         in ans_str
+            or "failed"       in ans_str
+        ):
             for key, val in self.rag_fallbacks.items():
                 tokens = re.findall(r"\w+", key.lower())
                 tokens = [t for t in tokens if len(t) > 2]
-                if tokens and all(tok in q_low for tok in tokens):
+                if tokens and all(tok in ans_str or tok in question.lower() for tok in tokens):
                     return val
 
         return raw_ans