import gradio as gr
from mailparser import parse_from_string
from bs4 import BeautifulSoup

def accept_mail(name):
    email = parse_from_string(name)
    return email

def clean_email(email):
    soup = BeautifulSoup(email.body, 'html.parser')

    for tag in soup.find_all(['style', 'link']):
        tag.decompose()

    cleaned_text = ' '.join(soup.get_text(separator=' ').split())
    return cleaned_text

def present(email):
    email_info = {
        "Subject": email.subject,
        "From": email.from_,
        "To": email.to,
        "Date": email.date,
        "Message ID": email.message_id,
        "Headers": email.headers,
        "Attachments": email.attachments
    }
    return [
        email_info["Subject"], 
        email_info["From"], 
        email_info["To"], 
        email_info["Date"], 
        email_info["Message ID"], 
        email_info["Headers"], 
        email_info["Attachments"]
    ]

demo = gr.Interface(
    fn=present, 
    inputs="text", 
    outputs=["text", "text", "text", "text", "text", "text", "text"], 
    title="Email Info", 
    description="Enter the email content below to view its details."
)
demo.launch()