Pycrolis commited on
Commit
c3c0ed5
·
1 Parent(s): 28960ba

feat(agent): include file attachment links in questions before processing

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -78,11 +78,13 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
78
  for item in questions_data:
79
  task_id = item.get("task_id")
80
  question_text = item.get("question")
 
81
  if not task_id or question_text is None:
82
  print(f"Skipping item with missing task_id or question: {item}")
83
  continue
84
  try:
85
- submitted_answer = agent(question_text)
 
86
  answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
87
  results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
88
  except Exception as e:
@@ -173,6 +175,14 @@ with gr.Blocks() as demo:
173
  outputs=[status_output, results_table]
174
  )
175
 
 
 
 
 
 
 
 
 
176
  if __name__ == "__main__":
177
  print("\n" + "-"*30 + " App Starting " + "-"*30)
178
  # Check for SPACE_HOST and SPACE_ID at startup for information
 
78
  for item in questions_data:
79
  task_id = item.get("task_id")
80
  question_text = item.get("question")
81
+ file_name = item.get("file_name")
82
  if not task_id or question_text is None:
83
  print(f"Skipping item with missing task_id or question: {item}")
84
  continue
85
  try:
86
+ question_with_attachment = compute_question_with_attachment(question_text, task_id, file_name)
87
+ submitted_answer = agent(question_with_attachment)
88
  answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
89
  results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
90
  except Exception as e:
 
175
  outputs=[status_output, results_table]
176
  )
177
 
178
+
179
+ def compute_question_with_attachment(question: str, task_id: str, file_name: str) -> str:
180
+ if file_name:
181
+ return f"{question}\n\nAttached file: https://agents-course-unit4-scoring.hf.space/files/{task_id}"
182
+ else:
183
+ return question
184
+
185
+
186
  if __name__ == "__main__":
187
  print("\n" + "-"*30 + " App Starting " + "-"*30)
188
  # Check for SPACE_HOST and SPACE_ID at startup for information