revert

app.py

@@ -2,8 +2,8 @@ import gradio as gr
 from mailparser import parse_from_string
 from bs4 import BeautifulSoup
 
-def accept_mail(name):
-    email = parse_from_string(name)
+def accept_mail(email_content):
+    email = parse_from_string(email_content)
     return email
 
 def clean_email(email):
@@ -22,36 +22,26 @@ def present(email_content):
         "To": email.to,
         "Date": email.date,
         "Message ID": email.message_id,
-        "Headers": email.headers,
-        "Attachments": email.attachments,
+        "Headers": str(email.headers),  # Convert dictionary to string for display
+        "Attachments": str(email.attachments),  # Convert list to string for display
         "Cleaned Body": cleaned_text
     }
     return [email_info[key] for key in email_info]
 
-with gr.Blocks() as demo:
-    gr.Markdown("## Email Info")
-    gr.Markdown("Enter the email content below to view its details.")
-    with gr.Row():
-        with gr.Column():
-            input_text = gr.Textbox(label="Email Content")
-            output_texts = [
-                gr.Textbox(label="Subject"),
-                gr.Textbox(label="From"),
-                gr.Textbox(label="To"),
-                gr.Textbox(label="Date"),
-                gr.Textbox(label="Message ID"),
-                gr.Textbox(label="Headers"),
-                gr.Textbox(label="Attachments"),
-                gr.Textbox(label="Cleaned Body", visible=False)  # Initially hidden
-            ]
-            button = gr.Button("Process Email")
-        with gr.Column():
-            output_cleaned_text = gr.Textbox(label="Cleaned Email Body")
-    
-    button.click(
-        present, 
-        inputs=input_text, 
-        outputs=output_texts + [output_cleaned_text]
-    )
-
+demo = gr.Interface(
+    fn=present, 
+    inputs="text", 
+    outputs=[
+        gr.components.Textbox(label="Subject"),
+        gr.components.Textbox(label="From"),
+        gr.components.Textbox(label="To"),
+        gr.components.Textbox(label="Date"),
+        gr.components.Textbox(label="Message ID"),
+        gr.components.Textbox(label="Headers"),
+        gr.components.Textbox(label="Attachments"),
+        gr.components.Textbox(label="Cleaned Body")
+    ],
+    title="Email Info",
+    description="Enter the email content below to view its details."
+)
 demo.launch()