Update app.py

app.py

@@ -187,11 +187,12 @@ class AgentState(TypedDict):
 # -------------------------
 
 class BasicAgent:
-    def __init__(self):
+    def __init__(self, session: requests.Session):
         if not OPENAI_API_KEY:
             raise EnvironmentError("OPENAI_API_KEY not set")
         self.llm = OpenAI(api_key=OPENAI_API_KEY)
         self.workflow = self._build_workflow()
+        self.session = session
 
     def _call_llm(self, prompt: str, max_tokens: int = 256) -> str:
         try:
@@ -248,10 +249,7 @@ class BasicAgent:
         if state["file_url"]:
             try:
                 print(f"Downloading {state['file_url']} …")
-                headers = {}
-                if HF_TOKEN:
-                    headers["Authorization"] = f"Bearer {HF_TOKEN}"
-                response = requests.get(state["file_url"], headers=headers, timeout=30)
+                response = self.session.get(state["file_url"], timeout=30)
                 response.raise_for_status()
                 data = response.content
                 print(f"Successfully downloaded file, size: {len(data)} bytes")
@@ -343,10 +341,13 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     questions_url = f"{api_url}/questions"
     submit_url = f"{api_url}/submit"
 
+    # Create a persistent session for all requests
+    sess = requests.Session()
+
     # 1. Instantiate Agent
     try:
         print("Initializing agent...")
-        agent = BasicAgent()
+        agent = BasicAgent(session=sess)  # Pass session to agent
         print("Agent initialized successfully.")
     except Exception as e:
         print(f"Error instantiating agent: {e}")
@@ -355,7 +356,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     # 2. Fetch Questions
     print(f"Fetching questions from: {questions_url}")
     try:
-        response = requests.get(questions_url, timeout=30)
+        response = sess.get(questions_url, timeout=30)
         response.raise_for_status()
         questions_data = response.json()
         if not questions_data:
@@ -419,7 +420,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     }
 
     try:
-        response = requests.post(submit_url, json=submission_data, timeout=60)
+        response = sess.post(submit_url, json=submission_data, timeout=60)
         response.raise_for_status()
         result_data = response.json()
         final_status = (