Update app.py

app.py

@@ -70,49 +70,92 @@ def get_verification_code(driver, email, max_retries=10, delay=2):
 
 def generate_random_email():
     """生成随机邮箱地址"""
-    random_prefix = ''.join(random.choices(string.ascii_letters + string.digits, k=8))
-    email = f"{random_prefix}@youxiang.dev"
-    logger.info(f"生成随机邮箱: {email}")
-    return {"email": email}
+    try:
+        random_prefix = ''.join(random.choices(string.ascii_letters + string.digits, k=8))
+        email = f"{random_prefix}@youxiang.dev"
+        logger.info(f"生成随机邮箱: {email}")
+        return {"status": "success", "email": email}
+    except Exception as e:
+        logger.error(f"生成邮箱失败: {str(e)}")
+        return {"status": "error", "message": str(e)}
 
 def get_email_code(email: str):
     """获取指定邮箱的验证码"""
+    if not email:
+        return {"status": "error", "message": "邮箱地址不能为空"}
+        
     logger.info(f"开始获取邮箱验证码: {email}")
+    driver = None
+    
     try:
         driver = setup_driver()
         code = get_verification_code(driver, email)
-        driver.quit()
         
         if code:
             logger.info(f"成功获取验证码: {code}")
-            return {"code": code}
+            return {"status": "success", "code": code}
         else:
             logger.error("未能获取验证码")
-            return {"error": "未能获取验证码"}
+            return {"status": "error", "message": "未能获取验证码"}
             
     except Exception as e:
         logger.error(f"获取验证码过程出错: {str(e)}")
-        return {"error": str(e)}
+        return {"status": "error", "message": str(e)}
     finally:
-        try:
-            driver.quit()
-        except:
-            pass
+        if driver:
+            try:
+                driver.quit()
+            except:
+                pass
 
-# Gradio 接口
-demo = gr.Interface(
-    [generate_random_email, get_email_code],
-    [
-        None,  # generate_random_email 不需要输入
-        gr.Textbox(label="邮箱地址")  # get_email_code 需要邮箱输入
-    ],
-    [
-        gr.JSON(label="生成的邮箱"),  # 返回 JSON 格式的邮箱
-        gr.JSON(label="验证码结果")   # 返回 JSON 格式的验证码
-    ],
-    title="邮箱验证码服务",
-    description="提供随机邮箱生成和验证码获取服务"
-)
+# 创建 Gradio 界面
+with gr.Blocks(title="邮箱验证码服务") as demo:
+    gr.Markdown("""
+    # 邮箱验证码服务
+    
+    提供两个主要功能：
+    1. 生成随机临时邮箱
+    2. 获取邮箱验证码
+    """)
+    
+    with gr.Tab("生成随机邮箱"):
+        with gr.Row():
+            email_btn = gr.Button("生成邮箱", variant="primary")
+            email_output = gr.JSON(label="结果")
+        
+        gr.Examples(
+            examples=[],
+            inputs=[],
+            outputs=[email_output],
+            fn=generate_random_email,
+            cache_examples=True,
+        )
+    
+    with gr.Tab("获取验证码"):
+        with gr.Row():
+            email_input = gr.Textbox(
+                label="邮箱地址",
+                placeholder="请输入要获取验证码的邮箱地址",
+                info="支持 @youxiang.dev 域名的邮箱"
+            )
+        with gr.Row():
+            code_btn = gr.Button("获取验证码", variant="primary")
+            code_output = gr.JSON(label="结果")
+            
+    # 绑定事件处理
+    email_btn.click(
+        fn=generate_random_email,
+        inputs=[],
+        outputs=email_output,
+        api_name="generate_email"
+    )
+    
+    code_btn.click(
+        fn=get_email_code,
+        inputs=email_input,
+        outputs=code_output,
+        api_name="get_code"
+    )
 
 # 启用队列处理请求
 demo.queue()