Create app.py

app.py

@@ -0,0 +1,67 @@
+import gradio as gr
+import nest_asyncio
+import asyncio
+import re
+import os
+
+from protonmail import ProtonMail
+
+nest_asyncio.apply()
+
+username = "laygyi5566@proton.me"
+password = "htz175039"
+
+proton = ProtonMail()
+
+# Only login and save session once
+if not os.path.exists("session.pickle"):
+    proton.login(username, password)
+    proton.save_session("session.pickle")
+
+# Load session
+proton.load_session("session.pickle", auto_save=True)
+
+
+async def fetch_verification_code():
+    messages = proton.get_messages()
+    if not messages:
+        return "❌ No messages found.", ""
+
+    message = proton.read_message(messages[0])
+    sender = message.sender.address
+    subject = message.subject
+    body = message.body
+
+    code_match = re.search(r"\b\d{6}\b", body)
+    if code_match:
+        verification_code = code_match.group()
+        result = f"✅ Verification code: {verification_code}"
+    else:
+        result = "❌ No verification code found."
+
+    # Save HTML (optional)
+    with open("message.html", "w", encoding="utf-8") as f:
+        f.write(body)
+
+    log = f"From: {sender}\nSubject: {subject}\n\n{body}"
+    return result, log
+
+
+def get_code_button_click():
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
+    result, log = loop.run_until_complete(fetch_verification_code())
+    return result, log
+
+
+# Gradio UI
+with gr.Blocks() as demo:
+    gr.Markdown("## ProtonMail Verification Code Fetcher")
+
+    get_code_button = gr.Button("Get Verification Code")
+    code_output = gr.Textbox(label="Verification Code", interactive=False)
+    log_output = gr.Textbox(label="Message Log", lines=15, interactive=False)
+
+    get_code_button.click(fn=get_code_button_click, outputs=[code_output, log_output])
+
+demo.launch()