Update app.py

app.py

@@ -9,20 +9,12 @@ from sklearn.cluster import KMeans
 import nltk
 import pandas as pd
 import smtplib
-from email.mime.multipart import MIMEMultipart
-from email.mime.text import MIMEText
-from email.mime.base import MIMEBase
-from email import encoders
 import os
-from sendgrid import SendGridAPIClient
-from sendgrid.helpers.mail import Mail, Attachment, FileContent, FileName, FileType, Disposition
-
 nltk.download('punkt')
 from nltk.tokenize import word_tokenize
 from nltk import tokenize
 import numpy as np
 import scipy.spatial
-import socket
 import csv
 
 
@@ -290,16 +282,16 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         diversity_slider = gr.Slider(0, 100, value=25, step=1, label="Diversity")
             
         submit_button = gr.Button("Submit")
+        output_message = gr.Textbox(label="Output Message", interactive=False)
         download_button = gr.File(label="Download Processed File", visible=False)
 
         def on_submit(llm, file, relevance, diversity):
             message, output_file = process_inputs(llm, file, relevance, diversity)
             if output_file:
-                return message, gr.File.update(value=output_file, visible=True)
-            else:
-                return message, gr.File.update(visible=False)
+                download_button.update(value=output_file, visible=True)
+            return message
 
-        submit_button.click(fn=on_submit, inputs=[llm_dropdown, file_upload, relevance_slider, diversity_slider], outputs=[None, download_button])
-            
+        submit_button.click(fn=on_submit, inputs=[llm_dropdown, file_upload, relevance_slider, diversity_slider], outputs=[output_message])
+   
 # Launch the Gradio app
 demo.launch()