Update app.py

app.py

@@ -9,7 +9,7 @@ import nltk
 from nltk.tokenize import sent_tokenize
 import torch
 from fastapi import FastAPI
-from fastapi.responses import RedirectResponse, FileResponse
+from fastapi.responses import RedirectResponse, FileResponse, JSONResponse
 from gtts import gTTS
 import tempfile
 import os
@@ -206,7 +206,7 @@ def summarize_document(file, summary_length: str, enable_tts: bool):
     try:
         summary = generate_summary(text, summary_length)
         audio_path = text_to_speech(summary) if enable_tts else None
-        pdf_path = create_pdf(summary, original_filename)
+        pdf_path = create_pdf(summary, original_filename) if summary else None
         return summary, "Summary complete", audio_path, pdf_path
     except Exception as e:
         return f"Summarization error: {str(e)}", "Error", None, None
@@ -243,6 +243,14 @@ with gr.Blocks(title="Document Summarizer", theme=gr.themes.Soft()) as demo:
     def toggle_audio_visibility(enable_tts):
         return gr.Audio(visible=enable_tts)
     
+    def update_ui(summary, status, audio_path, pdf_path):
+        return (
+            summary,
+            status,
+            gr.Audio(visible=audio_path is not None, value=audio_path),
+            gr.File(visible=pdf_path is not None, value=pdf_path)
+        )
+    
     tts_checkbox.change(
         fn=toggle_audio_visibility,
         inputs=tts_checkbox,
@@ -252,8 +260,11 @@ with gr.Blocks(title="Document Summarizer", theme=gr.themes.Soft()) as demo:
     submit_btn.click(
         fn=summarize_document,
         inputs=[file_input, length_radio, tts_checkbox],
-        outputs=[output, status, audio_output, pdf_download],
-        api_name="summarize"
+        outputs=[output, status, audio_output, pdf_download]
+    ).then(
+        fn=update_ui,
+        inputs=[output, status, audio_output, pdf_download],
+        outputs=[output, status, audio_output, pdf_download]
     )
 
 # FastAPI endpoints for files
@@ -269,4 +280,4 @@ app = gr.mount_gradio_app(app, demo, path="/")
 
 @app.get("/")
 def redirect_to_interface():
-    return RedirectResponse(url="/")
+    return RedirectResponse(url="/")