Update app.py

app.py

@@ -12,30 +12,36 @@ from openai import OpenAI
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 OPENAI_FALLBACK_KEY = "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"  # <<< Replace with your actual key here
 
-# --- OpenAI Agent with fallback key ---
+# --- OpenAI Agent with enhanced formatting and logic ---
 class OpenAIAgent:
     def __init__(self, model="gpt-4"):
         self.model = model
         self.system_prompt = (
             "You are a precise assistant that always returns a single, direct answer, with no explanation or justification. "
-            "If the task is to list, format your output exactly as instructed (e.g., comma-separated, sorted, etc.). "
-            "For tasks with media (images, video), assume a workaround or describe the answer based on the question text. "
-            "Never reply with disclaimers like 'I'm a text-based AI'."
+            "If the task asks for a list, format it exactly as instructed (e.g., comma-separated, alphabetized). "
+            "If there is a file missing, infer the answer using the question context. "
+            "Avoid all disclaimers, apologies, or messages about being a language model."
         )
-        
-        # Use env key or fallback
         api_key = os.getenv("OPENAI_API_KEY") or OPENAI_FALLBACK_KEY
         if not api_key or "sk-" not in api_key:
             raise ValueError("A valid OpenAI API key must be set in the environment or OPENAI_FALLBACK_KEY.")
         self.client = OpenAI(api_key=api_key)
 
-    def __call__(self, question: str, task_id: str = None) -> str:
+    def __call__(self, question: str, task_id: str = None, file_name: str = "") -> str:
         context = ""
-        if task_id:
+        if task_id and file_name:
             try:
                 context = self._fetch_file_context(task_id)
             except Exception as e:
                 print(f"Warning: Failed to fetch or parse file for task {task_id}: {e}")
+        # Add custom cues based on the question content
+        if "comma separated" in question.lower():
+            question = "Format your answer as a comma-separated list in alphabetical order.\n\n" + question
+        if "only list the ingredients" in question.lower():
+            question = "Output just the ingredient names, comma-separated.\n\n" + question
+        if "numeric output" in question.lower():
+            question = "Only provide the final numeric result, no units or explanation.\n\n" + question
+
         user_input = f"{question}\n\nAdditional context:\n{context}" if context else question
         return self._call_openai(user_input)
 
@@ -71,10 +77,10 @@ class OpenAIAgent:
 
             if extension == ".pdf":
                 return self._parse_pdf(tmp_file.name)
-            elif extension in [".txt", ".csv"]:
+            elif extension in [".txt", ".csv", ".py"]:
                 return response.text
             else:
-                return f"[Non-text file of type {extension}, cannot parse.]"
+                return f"[Non-text file of type {extension}, not processed.]"
 
     def _parse_pdf(self, filepath):
         doc = fitz.open(filepath)
@@ -114,10 +120,11 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     for item in questions_data:
         task_id = item.get("task_id")
         question_text = item.get("question")
+        file_name = item.get("file_name", "")
         if not task_id or question_text is None:
             continue
         try:
-            submitted_answer = agent(question_text, task_id=task_id)
+            submitted_answer = agent(question_text, task_id=task_id, file_name=file_name)
             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})
         except Exception as e: