Update app.py

app.py

@@ -10,10 +10,22 @@ from openinference.instrumentation.smolagents import SmolagentsInstrumentor
 
 from groq_api import GrokApi
 
+My_Agent = MultiAgent
 
 
-My_Agent = MultiAgent
+import json
+
+CACHE_FILE = "answers_cache.json"
 
+def load_cached_answers():
+    if os.path.exists(CACHE_FILE):
+        with open(CACHE_FILE, "r") as f:
+            return json.load(f)
+    return {}
+
+def save_cached_answers(cache):
+    with open(CACHE_FILE, "w") as f:
+        json.dump(cache, f)
 
 
 
@@ -36,6 +48,9 @@ def run_and_submit_all(nb_questions: int, profile: gr.OAuthProfile | None):
     Fetches all questions, runs my Agent on them, submits all answers,
     and displays the results.
     """
+
+    cached_answers = load_cached_answers()
+    
     # --- Determine HF Space Runtime URL and Repo URL ---
     space_id = os.getenv("SPACE_ID") # Get the SPACE_ID for sending link to the code   
 
@@ -90,8 +105,12 @@ def run_and_submit_all(nb_questions: int, profile: gr.OAuthProfile | None):
     # for testing keep only some questions
     questions_data = questions_data[:nb_questions]
 
-    print(f"Running agent on {len(questions_data)} questions...")
-    for item in questions_data:
+    new_questions = [q for q in questions_data if q["task_id"] not in cached_answers]
+
+    answers_payload_new = []
+
+    print(f"Running agent on {len(new_questions)} questions...")
+    for item in new_questions:
         task_id = item.get("task_id")
         question_text = item.get("question")
         file_name = item.get("file_name")
@@ -107,13 +126,20 @@ def run_and_submit_all(nb_questions: int, profile: gr.OAuthProfile | None):
                 agent_question += f"\n\nFile URL: {file_question_url}"
 
             submitted_answer = agent(agent_question)
+            answers_payload_new.append({"task_id": task_id, "submitted_answer": submitted_answer})
 
-            answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
+            cached_answers[task_id] = submitted_answer
+
+            #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:
              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}"})
 
+    save_cached_answers(cached_answers)
+    answers_payload = [{"task_id": k, "submitted_answer": v} for k, v in cached_answers.items()]
+
+    
     if not answers_payload:
         print("Agent did not produce any answers to submit.")
         return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)