Update app.py

app.py

@@ -24,6 +24,43 @@ from typing import List, Dict, Any, Optional
 from litellm import completion
 from duckduckgo_search import DDGS
 
+# 用於獲取並解析附件
+def fetch_and_summarize_attachment(task_id, question, files_api_url="https://agents-course-unit4-scoring.hf.space/files/"):
+    import pandas as pd, requests
+    q_lower = question.lower()
+    keywords = ["attached", "file", "excel", "csv", "spreadsheet", "audio", "mp3", "image", "picture", "python code", "code"]
+    if not any(k in q_lower for k in keywords):
+        return ""
+    try:
+        file_url = f"{files_api_url}{task_id}"
+        response = requests.get(file_url, timeout=10)
+        local_path = f"/tmp/{task_id}"
+        with open(local_path, "wb") as f:
+            f.write(response.content)
+        if local_path.endswith(".csv") or b"," in response.content[:512]:
+            df = pd.read_csv(local_path)
+            summary = df.head(10).to_string()
+            return f"[Attachment Detected: CSV Table]\nFirst 10 rows:\n{summary}\n"
+        elif local_path.endswith(".xls") or local_path.endswith(".xlsx"):
+            df = pd.read_excel(local_path)
+            summary = df.head(10).to_string()
+            return f"[Attachment Detected: Excel Table]\nFirst 10 rows:\n{summary}\n"
+        elif local_path.endswith(".txt") or b"\n" in response.content[:512]:
+            with open(local_path, "r", encoding="utf-8", errors="ignore") as f:
+                text = f.read(1000)
+            return f"[Attachment Detected: Text File]\n{text}\n"
+        elif local_path.endswith(".py"):
+            with open(local_path, "r", encoding="utf-8", errors="ignore") as f:
+                code = f.read(1000)
+            return f"[Attachment Detected: Python Code]\n{code}\n"
+        else:
+            return "[Attachment Detected, but file type not supported for summary]\n"
+    except Exception as e:
+        return f"[Attachment Error: {e}]\n"
+
+
+
+
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 DEFAULT_USERNAME = "lethaq"  # 默认用户名，您可以修改为您的用户名
@@ -52,6 +89,11 @@ class DuckDuckGoSearchTool:
         except Exception as e:
             print(f"DuckDuckGo search error: {e}")
             return [{"title": f"Search error: {e}", "body": "", "href": ""}]
+
+
+
+            
+
     
     def __call__(self, query: str, max_results: int = 8) -> Dict[str, Any]:
         """
@@ -443,18 +485,25 @@ def run_and_submit_all():
     for item in questions_data:
         task_id = item.get("task_id")
         question_text = item.get("question")
-        if not task_id or question_text is None:
-            print(f"Skipping item with missing task_id or question: {item}")
-            continue
-        try:
-            print(f"Processing task {task_id}: {question_text[:50]}...")
-            submitted_answer = agent(question_text)
-            answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
-            results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
-            print(f"Answer for task {task_id}: {submitted_answer[:50]}...")
-        except Exception as e:
-            print(f"Error running agent on task {task_id}: {e}")
-            results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"AGENT ERROR: {e}"})
+    if not task_id or question_text is None:
+        print(f"Skipping item with missing task_id or question: {item}")
+        continue
+    try:
+        # ① Attachment 摘要插入
+        attachment_context = fetch_and_summarize_attachment(task_id, question_text)
+        # ② 拼 prompt（根據你的 agent 用法）：
+        # （方案A）如果 agent 支援 attachment_context 參數
+        # submitted_answer = agent(question_text, attachment_context=attachment_context)
+        # （方案B）最直接：把 attachment 拼到問題開頭
+        full_question = (attachment_context + "\n" + question_text).strip()
+        submitted_answer = agent(full_question)
+        answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
+        results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
+        print(f"Answer for task {task_id}: {submitted_answer[:50]}...")
+    except Exception as e:
+        print(f"Error running agent on task {task_id}: {e}")
+        results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"AGENT ERROR: {e}"})
+
 
     if not answers_payload:
         print("Agent did not produce any answers to submit.")