Update agent.py

agent.py

@@ -79,35 +79,55 @@ class Agent:
         file_names = input_data.get("file_names", [])
         task_id = input_data.get("task_id", "")
 
+        system_prompt = (
+            "You are a member of a multidisciplinary research institute, tackling complex and ambiguous problems across knowledge, reasoning, and vision.\n\n"
+            "You have access to tools like search engines, calculators, and data analysis environments. Your task is to solve the following question carefully and completely.\n\n"
+            "You must:\n"
+            "- Think step by step, and write down all reasoning.\n"
+            "- If information is missing, use what you know and search if needed.\n"
+            "- If you encounter a file, inspect its content and extract relevant information.\n"
+            "- Use available tools only when needed, but do not rely on them blindly.\n"
+            "- If a tool does not return the final answer, analyze the result and continue reasoning.\n\n"
+            "Always:\n"
+            "- Confirm that your answer satisfies the constraints (e.g., format, brevity, units).\n"
+            "- Answer in one English sentence only, with no explanation.\n"
+            "- If the question has a strict required output format, follow it exactly.\n"
+            "- Do not end your output until you're confident your answer is final and complete.\n\n"
+            "---\n\n"
+            "Now solve the following task as best as possible. Do not skip steps. Think hard. Use all your skills and tools. Good luck.\n\n"
+        )
+
+        file_summary = ""
         try:
             if file_names:
                 file_path = f"/home/user/app/files/{task_id}/{file_names[0]}"
                 ext = Path(file_path).suffix.lower()
                 if ext in [".csv", ".tsv"]:
                     df = pd.read_csv(file_path)
-                    return f"Loaded table with {df.shape[0]} rows and {df.shape[1]} columns."
-                elif ext in [".xlsx"]:
+                    file_summary = f"The following table has been loaded with {df.shape[0]} rows and {df.shape[1]} columns:\n{df.head(3).to_string(index=False)}"
+                elif ext == ".xlsx":
                     df = pd.read_excel(file_path)
-                    return f"Loaded spreadsheet with {df.shape[0]} rows and {df.shape[1]} columns."
+                    file_summary = f"The following spreadsheet has been loaded with {df.shape[0]} rows and {df.shape[1]} columns:\n{df.head(3).to_string(index=False)}"
                 elif ext in [".json", ".jsonl"]:
                     with open(file_path, "r", encoding="utf-8") as f:
                         if ext == ".jsonl":
                             data = [json.loads(line) for line in f if line.strip()]
                         else:
                             data = json.load(f)
-                    return f"Loaded JSON with {len(data)} items."
+                    file_summary = f"The following JSON data was loaded ({len(data)} items)."
                 elif ext == ".docx":
                     doc = Document(file_path)
                     text = "\n".join([para.text for para in doc.paragraphs])
-                    return f"Loaded DOCX with {len(text)} characters."
+                    file_summary = f"Extracted text from DOCX ({len(text)} characters)."
                 elif ext == ".pdf":
                     doc = fitz.open(file_path)
                     text = "".join([page.get_text() for page in doc])
-                    return f"Loaded PDF with {len(doc)} pages and {len(text)} characters."
+                    file_summary = f"Extracted text from PDF ({len(doc)} pages, {len(text)} characters)."
                 else:
-                    return "Unsupported file type for this task."
+                    file_summary = "(Unsupported file type — skipping file content.)"
 
-            result = self.agent.run(question)
+            full_prompt = system_prompt + file_summary + f"\n\nTASK:\n{question}"
+            result = self.agent.run(full_prompt)
             return result.strip()
         except Exception as e:
             return f"AGENT ERROR: {str(e)}"