Update app.py

app.py

@@ -147,64 +147,29 @@ def answer_question(prompt):
     return generated_answer
 
     
-def process_inputs(llm, file, relevance, diversity, email):
+def process_inputs(llm, file, relevance, diversity):
     # Check if file is uploaded
     if file is not None:
         # Read questions from the uploaded Excel file
         try:
             df = pd.read_excel(file.name, engine='openpyxl')
         except Exception as e:
-            return f"Failed to read Excel file: {e}"
+            return f"Failed to read Excel file: {e}", None
 
         # Ensure that there is only one column in the file
         if df.shape[1] != 1:
-            return "The uploaded file must contain only one column of questions."
+            return "The uploaded file must contain only one column of questions.", None
 
         questions_list = df.iloc[:, 0].tolist()
     else:
-        return "No questions provided."
-
-    # Save questions to a CSV file
-    df = pd.DataFrame(questions_list, columns=["Questions"])
-    csv_file = "questions.csv"
-    df.to_csv(csv_file, index=False)
-
-    # Check network connectivity to the custom port
-    try:
-        socket.create_connection(("your.relay.server", 25), timeout=10)  # Replace with your relay server and port
-    except OSError:
-        return "Network is unreachable. Unable to send email."
-
-    # Email the CSV file
-    sender_email = "auditllms@gmail.com"
-    sender_password = "opri fcxx crkh bvfj"
-    receiver_email = email
-    subject = "Your Submitted Questions"
-    body = "Thank you for your submission. Please find attached the CSV file containing your questions."
-
-    message = MIMEMultipart()
-    message['From'] = sender_email
-    message['To'] = receiver_email
-    message['Subject'] = subject
-    message.attach(MIMEText(body, 'plain'))
-
-    with open(csv_file, "rb") as attachment:
-        part = MIMEBase('application', 'octet-stream')
-        part.set_payload(attachment.read())
-        encoders.encode_base64(part)
-        part.add_header('Content-Disposition', f"attachment; filename= questions.csv")
-        message.attach(part)
-
-    try:
-        with smtplib.SMTP('your.relay.server', 25) as server:  # Use your relay server and custom port
-            server.starttls()  # Upgrade the connection to a secure encrypted SSL/TLS connection
-            server.login(sender_email, sender_password)
-            server.sendmail(sender_email, receiver_email, message.as_string())
-    except Exception as e:
-        return f"Failed to send email: {e}"
-
-    return "Submitted"
+        return "No questions provided.", None
 
+    # Save questions to a new Excel file
+    output_df = pd.DataFrame(questions_list, columns=["Questions"])
+    output_file = "processed_questions.xlsx"
+    output_df.to_excel(output_file, index=False)
+
+    return "Processing complete. Download the file below.", output_file
 
     
 text_list = []
@@ -313,15 +278,19 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         gr.Markdown("Upon completion, please provide your email address. We will compile and send the answers to you promptly.")
             
         llm_dropdown = gr.Dropdown([("Llama", "TheBloke/Llama-2-7B-Chat-GGML"), ("Falcon", "TheBloke/Falcon-180B-Chat-GGUF"), ("Zephyr", "TheBloke/zephyr-quiklang-3b-4K-GGUF"), ("Vicuna", "TheBloke/vicuna-33B-GGUF"), ("Claude", "TheBloke/claude2-alpaca-13B-GGUF"), ("Alpaca", "TheBloke/LeoScorpius-GreenNode-Alpaca-7B-v1-GGUF")], label="Large Language Model")
-        file_upload = gr.File(label="Upload a File with Questions", file_types=["csv"])
+        file_upload = gr.File(label="Upload a File with Questions", file_types=["xlsx"])
         relevance_slider = gr.Slider(0, 100, value=70, step=1, label="Relevance")
         diversity_slider = gr.Slider(0, 100, value=25, step=1, label="Diversity")
-        email_input = gr.Textbox(label="Enter your email address", placeholder="name@example.com")
             
         submit_button = gr.Button("Submit")
         output_textbox = gr.Textbox(label="Output")
-        submit_button.click(fn=process_inputs, inputs=[llm_dropdown, file_upload, relevance_slider, diversity_slider, email_input], outputs=output_textbox)
+        download_button = gr.File(label="Download Processed File")
+
+        def on_submit(llm, file, relevance, diversity):
+            result, output_file = process_inputs(llm, file, relevance, diversity)
+            return result, output_file
 
+        submit_button.click(fn=on_submit, inputs=[llm_dropdown, file_upload, relevance_slider, diversity_slider], outputs=[output_textbox, download_button])
 
     
 # Launch the Gradio app