new gradio

app.py

@@ -373,16 +373,22 @@ def format_quiz_for_display(results):
     return "\n".join(output)
 
 
-
 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, results
+        json_path = "analysis_results.json"
+        txt_path = "analysis_results.txt"
+        with open(json_path, "w", encoding="utf-8") as f:
+            json.dump(results, f, indent=2, ensure_ascii=False)
+        with open(txt_path, "w", encoding="utf-8") as f:
+            f.write(formatted_output)
+        
+        return formatted_output, json_path, txt_path
     except Exception as e:
         error_msg = f"Error processing document: {str(e)}"
-        return error_msg, None
+        return error_msg, None, None
 
 with gr.Blocks(title="Quiz Generator") as app:
     gr.Markdown("# Quiz Generator")
@@ -406,49 +412,13 @@ 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)}
+            json_file_output = gr.File(label="Download JSON")
+            txt_file_output = gr.File(label="Download TXT")
     
-    # 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_and_store,
+        fn=analyze_document,
         inputs=[input_text, api_key],
-        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")]
+        outputs=[output_results, json_file_output, txt_file_output]
     )
 
 if __name__ == "__main__":