Update app.py

app.py

@@ -41,13 +41,14 @@ def summary(input, summary_type="medium"):
     output = text_summary(input, max_length=max_output_tokens, min_length=min_output_tokens, truncation=True)
     return output[0]['summary_text']
 
-# Gradio interface
-def save_output(summary_text):
+# Function to save the output summary to a file
+def save_summary(summary_text):
     """Save the summarized text to a file."""
     with open("summary_output.txt", "w") as file:
         file.write(summary_text)
     return "Summary saved to 'summary_output.txt'."
 
+# Gradio interface setup
 gr.close_all()
 
 # Create the Gradio interface
@@ -59,7 +60,7 @@ demo = gr.Interface(
     ],
     outputs=[
         gr.Textbox(label="SUMMARIZED TEXT", lines=10, placeholder="Your summarized text will appear here."),
-        gr.Button("Download Summary")
+        gr.Button("Save Summary", click=save_summary)
     ],
     title="PAVISHINI @ GenAI Project 1: Text Summarizer",
     description=(
@@ -71,4 +72,3 @@ demo = gr.Interface(
 
 demo.launch()
 
-