Update app.py

app.py

@@ -55,7 +55,8 @@ async def check_n_load_attach(session: aiohttp.ClientSession, task_id: str, ques
         print(f"Error downloading attachment for task {task_id}: {str(e)}")
         return None
 
-async def determine_extension(content_type: str, content: bytes, question: str) -> Optional[str]:
+async def determine_extension(content_type: str, content: bytes, question
+: str) -> Optional[str]:
     # Check if the question mentions Excel
     if "excel" in question.lower():
         # Check for XLS signature
@@ -211,13 +212,13 @@ async def run_and_submit_all(profile: gr.OAuthProfile | None):
 
             
             #select question we want to focus
-#            if "excel" in question_text.lower() or "mp3" in question_text.lower() or "chess" in question_text.lower():
+            if "excel" in question_text.lower() or "mp3" in question_text.lower() or "pitchers" in question_text.lower():
             file_path = await check_n_load_attach(session, task_id, question_text.lower())
             result = await process_question(agent, question_text, task_id, file_path, results_log)
             if result:
                 answers_payload.append(result)
-#            else: 
-#                print("Skipping unrelated question.")    
+            else: 
+                print("Skipping unrelated question.")    
 
         if not answers_payload:
             print("Agent did not produce any answers to submit.")