Update agent.py

agent.py

@@ -4,6 +4,7 @@ import requests
 import tempfile
 import pandas as pd
 from openai import OpenAI
+
 try:
     from duckduckgo_search import DDGS
 except ImportError:
@@ -38,7 +39,7 @@ class BasicAgent:
                 href = result.get('href', '')
                 formatted_results += f"{i}. {title}\n   URL: {href}\n   Description: {body}\n\n"
             return formatted_results
-        except Exception as e:
+        except Exception:
             return ""
 
     def excel_tool(self, file_url: str) -> str:
@@ -55,7 +56,7 @@ class BasicAgent:
                 return f"{round(total, 2)}"
             total = df.select_dtypes(include='number').sum().sum()
             return f"{round(total, 2)}"
-        except Exception as e:
+        except Exception:
             return ""
 
     def fetch_file_url(self, task_id):
@@ -70,7 +71,6 @@ class BasicAgent:
         return None
 
     def solve_chess_image(self, image_url: str) -> str:
-        """Uses GPT-4o Vision to answer a chess image question (requires image URL)."""
         prompt = (
             "You are a chess engine. Only answer with the best move for Black in algebraic notation (e.g., Qd1#). "
             "Do not explain your reasoning, do not include any commentary, only the move."
@@ -92,32 +92,31 @@ class BasicAgent:
                 temperature=0.0,
             )
             result = response.choices[0].message.content.strip()
-            # Remove commentary, keep only first move if any
             move = re.findall(r"\b([KQRNB]?[a-h]?[1-8]?x?[a-h][1-8](?:=[QRNB])?#?)\b", result)
             if move:
                 return move[0]
             return result
-        except Exception as e:
+        except Exception:
             return ""
 
     def __call__(self, question: str, task_id: str = None) -> str:
         file_url = self.fetch_file_url(task_id) if task_id else None
         file_result = None
 
-        # --- Chess image detection (heuristic: "chess", "move", "image", or "position") ---
+        # --- Chess image detection (priority) ---
         if file_url and ("chess" in question.lower() or "move" in question.lower() or "image" in question.lower() or "position" in question.lower()):
-            chess_result = self.solve_chess_image(file_url)
-            if chess_result and len(chess_result) < 10:  # likely algebraic notation
-                return chess_result
+            move = self.solve_chess_image(file_url)
+            if move and len(move) <= 6:
+                return move
 
-        # --- Excel heuristic (only try for likely Excel file) ---
+        # --- Excel file detection ---
         ext = file_url.split('.')[-1].lower() if file_url else ""
         if file_url and (ext in ["xlsx", "xls"] or "excel" in question.lower() or "spreadsheet" in question.lower()):
             file_result = self.excel_tool(file_url)
             if file_result and re.match(r'^\d+(\.\d+)?$', file_result):
                 return file_result
 
-        # --- Web search + LLM as before ---
+        # --- Web search + LLM ---
         search_snippet = self.web_search(question)
         prompt = PROMPT + f"\n\nWeb search results:\n{search_snippet}\n\nQuestion: {question}"
         response = self.llm.chat.completions.create(
@@ -133,7 +132,7 @@ class BasicAgent:
                 final_line = line.split(":", 1)[-1].strip(" .\"'")
                 break
 
-        # --- Fallback: Don't allow blank, placeholder, or apology answers ---
+        # --- Fallback for non-answers ---
         bads = [
             "", "unknown", "unable to determine", "unable to provide page numbers",
             "unable to access video content directly", "unable to analyze video content",
@@ -144,7 +143,6 @@ class BasicAgent:
         ]
         norm_final = (final_line or "").lower()
         if norm_final in bads or norm_final.startswith("unable") or norm_final.startswith("i'm sorry") or norm_final.startswith("i apologize"):
-            # Try to extract a plausible answer from web or file
             numbers = re.findall(r'\b\d{2,}\b', search_snippet)
             if numbers:
                 return numbers[0]
@@ -158,7 +156,6 @@ class BasicAgent:
                 file_words = re.findall(r'\b[A-Z][a-z]{2,}\b', str(file_result))
                 if file_words:
                     return file_words[0]
-            # --- Try to re-ask the LLM without apologies ---
             retry_prompt = (
                 "Based ONLY on the search results and/or file content above, return a direct answer to the question. "
                 "If you do not know, make your best plausible guess. Do NOT apologize or say you cannot assist. "