Update app.py

app.py

@@ -374,17 +374,18 @@ def format_quiz_for_display(results):
 
 
 
-def analyze_document(document_text: str, api_key: str) -> str:
+def analyze_document(document_text: str, api_key: str) -> tuple:
     os.environ["GOOGLE_API_KEY"] = api_key
     try:
         results = process_document_with_quiz(document_text)
         formatted_output = format_quiz_for_display(results)
-        return formatted_output
+        return formatted_output, results
     except Exception as e:
-        return f"Error processing document: {str(e)}"
+        error_msg = f"Error processing document: {str(e)}"
+        return error_msg, None
 
-with gr.Blocks(title="Quiz Generator ") as app:
-    gr.Markdown("Quiz Generator")
+with gr.Blocks(title="Quiz Generator") as app:
+    gr.Markdown("# Quiz Generator")
     
     with gr.Row():
         with gr.Column():
@@ -393,13 +394,11 @@ with gr.Blocks(title="Quiz Generator ") as app:
                 placeholder="Paste your document text here...",
                 lines=10
             )
-            
             api_key = gr.Textbox(
                 label="Gemini API Key",
                 placeholder="Enter your Gemini API key",
                 type="password"
             )
-            
             analyze_btn = gr.Button("Analyze Document")
         
         with gr.Column():
@@ -407,11 +406,49 @@ with gr.Blocks(title="Quiz Generator ") as app:
                 label="Analysis Results",
                 lines=20
             )
+            
+            with gr.Row():
+                json_btn = gr.Button("Download JSON")
+                txt_btn = gr.Button("Download TXT")
+    
+    # Create a state to store results
+    results_state = gr.State()
+    
+    # Function to handle analysis and store results in state
+    def analyze_and_store(text, key):
+        formatted, results = analyze_document(text, key)
+        return formatted, results
+    
+    # Function to prepare JSON for download
+    def prepare_json_download(results):
+        if results is None:
+            return None
+        return {"filename": "analysis_results.json", "content": json.dumps(results, indent=2, ensure_ascii=False)}
     
+    # Function to prepare TXT for download
+    def prepare_txt_download(text):
+        if not text or text.startswith("Error"):
+            return None
+        return {"filename": "analysis_results.txt", "content": text}
+    
+    # Connect the analyze button
     analyze_btn.click(
-        fn=analyze_document,
+        fn=analyze_and_store,
         inputs=[input_text, api_key],
-        outputs=output_results
+        outputs=[output_results, results_state]
+    )
+    
+    # Connect download buttons
+    json_btn.click(
+        fn=prepare_json_download,
+        inputs=[results_state],
+        outputs=[gr.File(label="Download JSON")]
+    )
+    
+    txt_btn.click(
+        fn=prepare_txt_download,
+        inputs=[output_results],
+        outputs=[gr.File(label="Download TXT")]
     )
 
 if __name__ == "__main__":