Update app.py

app.py

@@ -4,13 +4,11 @@ import json
 import os
 from smolagents import CodeAgent, tool
 from huggingface_hub import InferenceClient
-from agent import solve
 
-# Set up logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Answer cache (speeds up repeated evaluation)
+# Cache for answers
 CACHE_FILE = "answer_cache.json"
 if os.path.exists(CACHE_FILE):
     with open(CACHE_FILE) as f:
@@ -70,17 +68,13 @@ def web_search(query: str) -> str:
     except Exception as e:
         return f"Search error: {e}"
 
-# ---------- Custom model that wraps HF InferenceClient ----------
+# ---------- Custom model ----------
 class CustomHFModel:
     def __init__(self, model_id="HuggingFaceH4/zephyr-7b-beta"):
         self.client = InferenceClient(model=model_id, token=os.getenv("HF_TOKEN"))
         self.model_id = model_id
 
     def __call__(self, messages, **kwargs):
-        """
-        Expected by smolagents: takes a list of messages (e.g., [{"role": "user", "content": "..."}])
-        and returns the assistant's reply as a string.
-        """
         response = self.client.chat_completion(
             messages=messages,
             max_tokens=500,
@@ -98,12 +92,12 @@ try:
 except ImportError:
     logger.warning("duckduckgo-search not installed, web_search disabled.")
 
-model = CustomHFModel()  # you can change the model_id if desired
+model = CustomHFModel()
 agent = CodeAgent(tools=tools, model=model)
 
-# ---------- Main entry point ----------
-def answer_question(question: str) -> str:
-    """Called by the evaluator for each question."""
+# ---------- Main entry point (called by app.py) ----------
+def solve(question: str) -> str:
+    """This function must be named 'solve' because app.py imports it."""
     q_hash = hashlib.md5(question.encode()).hexdigest()
     if q_hash in answer_cache:
         logger.info(f"Cache hit for question: {question[:50]}...")
@@ -118,5 +112,4 @@ def answer_question(question: str) -> str:
 
     answer_cache[q_hash] = answer
     save_cache()
-    return answer
-    
+    return answer