+ download of csv

.gitignore

@@ -1,2 +1,4 @@
 env*
-.env*
+.env*
+review.csv
+*.csv

app.py

@@ -9,6 +9,7 @@ import asyncio
 import aiohttp
 from langfuse.langchain import CallbackHandler
 from langchain_core.messages import HumanMessage
+import tempfile
 
 # Initialize Langfuse CallbackHandler for LangGraph/Langchain (tracing)
 langfuse_handler = CallbackHandler()
@@ -55,7 +56,12 @@ async def generate_answers(profile: gr.OAuthProfile | None, progress=gr.Progress
         print(f"User logged in: {username}")
     else:
         print("User not logged in.")
-        return "Please Login to Hugging Face with the button.", None, gr.update(interactive=False), gr.update(value=0, visible=False)
+        return (
+            "Please Login to Hugging Face with the button.",
+            None,
+            gr.update(interactive=False),  # Disable submit button
+            gr.update(value=None, visible=False),  # Hide download button
+        )
     api_url = DEFAULT_API_URL
     questions_url = f"{api_url}/questions"
     try:
@@ -64,11 +70,21 @@ async def generate_answers(profile: gr.OAuthProfile | None, progress=gr.Progress
         questions_data = response.json()
         if not questions_data:
             print("Fetched questions list is empty.")
-            return "Fetched questions list is empty or invalid format.", None, gr.update(interactive=False), gr.update(value=0, visible=False)
+            return (
+                "Fetched questions list is empty or invalid format.",
+                None,
+                gr.update(interactive=False),
+                gr.update(value=None, visible=False),
+            )
         print(f"Fetched {len(questions_data)} questions.")
     except Exception as e:
         print(f"Error fetching questions: {e}")
-        return f"Error fetching questions: {e}", None, gr.update(interactive=False), gr.update(value=0, visible=False)
+        return (
+            f"Error fetching questions: {e}",
+            None,
+            gr.update(interactive=False),
+            gr.update(value=None, visible=False),
+        )
     agent = BasicAgent()
     results_log = []
     answers_payload = []
@@ -105,7 +121,23 @@ async def generate_answers(profile: gr.OAuthProfile | None, progress=gr.Progress
     cached_results_log = results_log
     progress(100, desc="Done.")
     results_df = pd.DataFrame(results_log)
-    return "Answer generation complete. Review and submit.", results_df, gr.update(interactive=True), gr.update(value=100, visible=True)
+
+    # Save answers to a temporary CSV so user can download
+    try:
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".csv", mode="w", newline="", encoding="utf-8") as tmp_csv:
+            results_df.to_csv(tmp_csv.name, index=False)
+            csv_path = tmp_csv.name
+            print(f"CSV saved to {csv_path}")
+    except Exception as e:
+        print(f"Failed to write CSV: {e}")
+        csv_path = None
+
+    return (
+        "Answer generation complete. Review and submit.",
+        results_df,
+        gr.update(interactive=True),  # Enable submit button
+        gr.update(value=csv_path, visible=bool(csv_path)),  # Show download button if csv written
+    )
 
 def submit_answers(profile: gr.OAuthProfile | None):
     """
@@ -187,17 +219,20 @@ with gr.Blocks() as demo:
     status_output = gr.Textbox(label="Status / Submission Result", lines=5, interactive=False)
     results_table = gr.DataFrame(label="Questions and Agent Answers", wrap=True)
 
+    # Download button appears after answers are generated
+    download_button = gr.DownloadButton(label="Download Answers CSV", visible=False)
+
     generate_button.click(
         fn=generate_answers,
         inputs=[],
-        outputs=[status_output, results_table, submit_button],
+        outputs=[status_output, results_table, submit_button, download_button],
         api_name="generate_answers"
     )
     submit_button.click(
         fn=submit_answers,
         inputs=[],
         outputs=[status_output, results_table],
-        api_name="submit_answers"
+        api_name="submit_answers",
     )
 
 if __name__ == "__main__":