Update app.py

app.py

@@ -15,6 +15,9 @@ from io import BytesIO
 import base64
 import threading
 from queue import Queue
+import logging
+
+logging.basicConfig(level=logging.INFO)
 
 def download_pdf(url, output_path):
     urllib.request.urlretrieve(url, output_path)
@@ -241,8 +244,10 @@ class WorkerThread(threading.Thread):
                 i, question = job
                 result = generate_answer(question, openAI_key)
                 self.results.put((i, result))
+                logging.info(f"Job {i} completed successfully.")
             except Exception as e:
                 self.results.put((i, str(e)))
+                logging.error(f"Error on job {i}: {str(e)}")
 
 if generate_all:
     questions = [st.session_state.get(f"question{i}", "") for i in range(row_count)]
@@ -252,19 +257,11 @@ if generate_all:
 
     workers = [WorkerThread(jobs, results) for _ in range(num_concurrent_calls)]
 
-    for worker in workers:
-        worker.start()
-
-    for i, question in enumerate(questions):
-        jobs.put((i, question))
-
-    for _ in range(num_concurrent_calls):
-        jobs.put(None)
-
     for worker in workers:
         worker.join()
 
-    # Collect all results first
+    logging.info("All worker threads have finished.")
+
     answers = {}
     while not results.empty():
         i, answer = results.get()
@@ -273,6 +270,9 @@ if generate_all:
         else:
             answers[i] = answer
 
-    # Update session state in the main thread
+    logging.info(f"Collected {len(answers)} answers.")
+
     for i, answer in answers.items():
         st.session_state[f'session_answer{i}'] = answer
+
+    logging.info("Session state updated with answers.")