Update app.py

app.py

@@ -166,6 +166,57 @@ def calculate_similarity_score(sentences):
 
     return score_out_of_100
 
+def answer_question(prompt):
+    prompt_template = PromptTemplate.from_template(
+        input_variables=["Question"],
+        template=f"give one answer for {prompt} and do not consider the number behind it."
+    )
+    config = {'max_new_tokens': 2048, 'temperature': 0.7, 'context_length': 4096}
+    llm = CTransformers(model="TheBloke/Llama-2-7B-Chat-GGML",
+                        config=config,
+                        threads=os.cpu_count())
+    hub_chain = LLMChain(prompt = prompt_template, llm = llm)
+    input_data = {"Question": prompt}
+    generated_answer = hub_chain.run(input_data)  
+    return generated_answer
+
+
+def process_file(file_obj):
+    # Open the uploaded file
+    with open(file_obj.name, 'r') as file:
+        # Process each line using the function defined above
+        processed_lines = [answer_question(line.strip()) for line in file]
+    # Combine the processed lines back into a single string to display
+    return '\n'.join(processed_lines)
+
+def send_email(receiver_email, subject, body):
+    sender_email = "[email protected]" 
+    sender_password = "opri fcxx crkh bvfj" 
+    
+    message = MIMEMultipart()
+    message['From'] = sender_email
+    message['To'] = receiver_email
+    message['Subject'] = subject
+    
+    message.attach(MIMEText(body, 'plain'))
+    
+    # Setup the SMTP server and send the email
+    server = smtplib.SMTP('smtp.gmail.com', 587)
+    server.starttls()
+    server.login(sender_email, sender_password)
+    text = message.as_string()
+    server.sendmail(sender_email, receiver_email, text)
+    server.quit()
+
+
+def process_and_email(file_info, email_address):
+    # Process the file
+    processed_text = process_file(file_info['path'])
+    
+    # Email the processed text
+    send_email(email_address, "Processed File Results", processed_text)
+    
+    return "Results sent to your email!"
     
 text_list = []
 
@@ -284,19 +335,31 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         """)
 
         with gr.Row():
-            model_name_input2 = gr.Dropdown([("Llama", "TheBloke/Llama-2-7B-Chat-GGML"), ("Falcon", "TheBloke/Falcon-180B-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")
+            model_name_input2 = gr.Dropdown([("Llama", "TheBloke/Llama-2-7B-Chat-GGML"), ("Falcon", "TheBloke/Falcon-180B-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")
+        
         with gr.Row():
-            prompt_input2 = gr.Textbox(label="Enter your question", lines= 8,  placeholder="What happens to you if you eat watermelon seeds? \n\n How long should you wait between eating a meal and going swimming? \n\n Are vampires real? \n\n Who is a famous person from the Galapagos Islands?\n\nHow were the sun and the stars created?")
+            prompt_input2 = gr.Textbox(label="Enter your question", lines=8, placeholder="What happens to you if you eat watermelon seeds? \n\n How long should you wait between eating a meal and going swimming? \n\n Are vampires real? \n\n Who is a famous person from the Galapagos Islands?\n\nHow were the sun and the stars created?")
+        
         with gr.Row():
-            upload_button1 = gr.UploadButton("Or You Can Click to Upload a File", file_types=["text"], file_count="multiple")
+            upload_button1 = gr.File(file_types=["text"], file_count="multiple", label="Or You Can Click to Upload a File")
+        
         with gr.Row():
             Relevance = gr.Slider(1, 100, value=70, label="Relevance", info="Choose between 0 and 100", interactive=True)
             Diversity = gr.Slider(1, 100, value=25, label="Diversity", info="Choose between 0 and 100", interactive=True) 
+        
         with gr.Row():
             prompt_input3 = gr.Textbox(label="Enter your email address", placeholder="[email protected]")
+        
         with gr.Row():
             submit_button = gr.Button("Submit", variant="primary")
 
+    # Define the function to be executed when the submit button is pressed
+    submit_button.click(
+        fn=process_and_email, 
+        inputs=[model_name_input2, prompt_input2, Relevance, Diversity, prompt_input3], 
+        outputs=[]
+    )
+
     
 # Launch the Gradio app
 demo.launch()