Update app.py

app.py

@@ -10,7 +10,7 @@ from typing import Optional
 import pytz
 from smolagents import CodeAgent, HfApiModel, load_tool, tool
 from tools.final_answer import FinalAnswerTool
-from tools.web_search     import DuckDuckGoSearchTool
+from tools.web_search import DuckDuckGoSearchTool
 
 # -----------------------------------------------------------------------------
 # 1) YOUR CUSTOM TOOLS (with proper docstrings for Args)
@@ -85,11 +85,10 @@ with open("prompts.yaml", "r") as f:
     prompt_templates = yaml.safe_load(f)
 
 # extract the fallback map from YAML
-rag_fallbacks = prompt_templates.get("rag_fallbacks", {})
+rag_fallbacks = prompt_templates.get("fallbacks", {})
 
 final_answer = FinalAnswerTool()
-web_search   = DuckDuckGoSearchTool()
-# If you need your text-to-image tool, uncomment:
+web_search = DuckDuckGoSearchTool()
 # image_generation = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 class BasicAgent:
@@ -123,21 +122,20 @@ class BasicAgent:
             raw_ans = self.agent.run(question)
         except Exception:
             raw_ans = ""
-        # 2) stringify for the empty/failure check
         ans_str = str(raw_ans)
 
-        # 3) if it failed to produce anything meaningful, try RAG fallback
-        if not ans_str.strip() or ans_str.lower().startswith("i'm sorry"):
+        # 2) determine if we need RAG fallback
+        is_empty      = not ans_str.strip()
+        is_sorry      = ans_str.lower().startswith("i'm sorry")
+        is_unanswered = any(marker in ans_str for marker in ["Unanswered", "failed", "None"])
+        if is_empty or is_sorry or is_unanswered:
             q_low = question.lower()
             for key, val in self.rag_fallbacks.items():
-                # extract tokens > 2 chars from the key
                 tokens = re.findall(r"\w+", key.lower())
                 tokens = [t for t in tokens if len(t) > 2]
-                # if all appear in the question, use this fallback
                 if all(tok in q_low for tok in tokens):
                     return val
 
-        # otherwise return the real answer
         return raw_ans
 
 # -----------------------------------------------------------------------------
@@ -186,7 +184,7 @@ def run_and_submit_all(profile: Optional[gr.OAuthProfile]):
         }
         r2 = requests.post(submit_url, json=payload, timeout=60)
         r2.raise_for_status()
-        res    = r2.json()
+        res = r2.json()
         status = (
             f"✔ Submission Successful!\n"
             f"User: {res.get('username')}\n"
@@ -210,11 +208,11 @@ with gr.Blocks() as demo:
     2. Click “Run Evaluation & Submit All Answers” below.
     """)
     gr.LoginButton()
-    btn    = gr.Button("Run Evaluation & Submit All Answers")
-    out_s  = gr.Textbox(label="Run Status / Submission Result", lines=5, interactive=False)
-    out_df = gr.DataFrame(label="Questions and Agent Answers", wrap=True)
+    btn   = gr.Button("Run Evaluation & Submit All Answers")
+    out_s = gr.Textbox(label="Run Status / Submission Result", lines=5, interactive=False)
+    out_df= gr.DataFrame(label="Questions and Agent Answers", wrap=True)
 
     btn.click(run_and_submit_all, [], [out_s, out_df])
 
 if __name__ == "__main__":
-    demo.launch(debug=True, share=False)
+    demo.launch(debug=True, share=False)