Upload folder using huggingface_hub

src/gradio_app.py

@@ -1,10 +1,11 @@
-import gradio as gr
-from langchain.chat_models import ChatOpenAI
-from src.summarization import summarize_wrapper
 import os
 
 import pypdfium2 as pdfium
-from PIL import Image
+import gradio as gr
+
+from langchain.chat_models import ChatOpenAI
+from src.summarization import summarize_wrapper
+from src.mailing import send_email
 
 # Function to render a specific page of a PDF file as an image
 def render_file(file):
@@ -19,13 +20,6 @@ def render_file(file):
         )
         images.append((bitmap.to_pil(), f"Seite {page_index+1}"))
 
-    # WORK FROM HERE TO RENDER THE COMPLETE PDF
-    # bitmap = pdf.render(
-    #     pdfium.PdfBitmap.to_pil,
-    #     page_indices=[0, 1, 2],
-    #     scale=300 / 72,  # 300dpi resolution
-    # )
-    # pil_image = bitmap.to_pil()
     return gr.update(
         value=images,
     )
@@ -75,24 +69,45 @@ def run_summarization_model_gradio(
         with gr.Row().style(equal_height=True):
             Header_box = generate_title(title=title, description=description)
         with gr.Row().style(equal_height=True):
-            clear = gr.Button("Clear")
+            clear = gr.Button("Clear All Components")
             file_upload = gr.File(
                 file_count="single",
                 file_types=[".pdf", ".txt"],
                 label="Upload PDF",
             )
-        with gr.Row().style(equal_height=True, height=1000):
+        with gr.Row().style(equal_height=True):
             summary_short = gr.Button("Kurze Zusammenfassung", interactive=False)
             summary_middle = gr.Button("Mittlere Zusammenfassung", interactive=False)
             summary_long = gr.Button("Lange Zusammenfassung", interactive=False)
         with gr.Row().style(equal_height=True):
             with gr.Column(scale=1):
-                summary_output = gr.Textbox(label="Zusammenfassung").style(
+                summary_output = gr.Textbox(label="Zusammenfassung", lines=9).style(
                     show_copy_button=True
                 )
             with gr.Column(scale=1):
                 show_pdf = gr.Gallery(label="Uploaded PDF").style(object_fit="contain")
 
+        with gr.Row().style(equal_height=True):
+            with gr.Column(scale=1):
+                recipiant_email = gr.Textbox(
+                    label="Recipiant Email", placeholder="Enter Email"
+                )
+                subject_email = gr.Textbox(label="Subject", placeholder="Enter Subject")
+                send_email_button = gr.Button("Open Email", interactive=False)
+            with gr.Column(scale=3):
+                email_instructions = gr.Textbox(
+                    label="Email Instructions",
+                    placeholder="Write Email Instructions here.",
+                    value=(
+                        "Dear Recipient\n\n"
+                        "Please find the summarization of the uploaded document below.\n\n"
+                        "<TEXT_FROM_LLM>\n\n"
+                        "Kind regards,\n"
+                        "Your Legal Assistant"
+                    ),
+                    lines=9,
+                )
+
         # Once a file is uploaded, enable the summarization buttons and visualize the uploaded file
         file_upload.upload(
             switch_buttons,
@@ -127,6 +142,11 @@ def run_summarization_model_gradio(
                 [gr.State(True)],
                 [summary_short, summary_middle, summary_long],
                 queue=False,
+            ).then(
+                switch_buttons,
+                [gr.State(True)],
+                [send_email_button, gr.State(None), gr.State(None)],
+                queue=False,
             )
 
         # The clear button clears the dashboard
@@ -137,6 +157,26 @@ def run_summarization_model_gradio(
             [gr.State(False)],
             [summary_short, summary_middle, summary_long],
             queue=False,
+        ).then(
+            lambda: None, None, show_pdf, queue=False
+        ).then(
+            lambda: None, None, send_email_button, queue=False
+        ).then(
+            lambda: None, None, email_instructions, queue=False
+        ).then(
+            lambda: None, None, recipiant_email, queue=False
+        )
+
+        # Email button click opens the default email client and fills in the email instructions
+        send_email_button.click(
+            send_email,
+            [
+                summary_output,
+                recipiant_email,
+                subject_email,
+                email_instructions,
+            ],
+            queue=False,
         )
 
     webui.queue()

src/mailing.py

@@ -0,0 +1,38 @@
+import webbrowser
+import urllib.parse
+from typing import List
+
+
+def send_email(
+    llm_text: str,
+    recipients: str = "",
+    subject: str = "",
+    email_wrapping_content: str = "<TEXT_FROM_LLM>",
+):
+    """Send an email to the specified recipients.
+
+    Args:
+        recipients (List[str], optional): List of email addresses. Defaults to [""].
+        subject (str): Subject of the email.
+        email_content (_type_): Content of the email.
+
+    Example usage:
+    - send_email("test@example.com;test2@example.com", "Hello!", "This is the body of the email.")
+    """
+    recipients_list = recipients.split(";")
+    email_addresses = ",".join([recipient.strip() for recipient in recipients_list])
+
+    # URL encode the subject and email content to ensure it's correctly parsed
+    subject = urllib.parse.quote(subject)
+
+    if "<TEXT_FROM_LLM>" in email_wrapping_content:
+        email_content = email_wrapping_content.replace("<TEXT_FROM_LLM>", llm_text)
+    else:
+        email_content = email_wrapping_content + "\n\n" + llm_text
+    email_content = urllib.parse.quote(email_content)
+
+    # Construct the mailto URL
+    mailto_url = f"mailto:{email_addresses}?subject={subject}&body={email_content}"
+
+    # Open the default email client
+    webbrowser.open(mailto_url)