Update app.py

app.py

@@ -234,28 +234,6 @@ spam_email = {
     "body": "CONGRATULATIONS! You have been selected as the winner of our international lottery! To claim your $5,000,000 prize, please send us your bank details and a processing fee of $100."
 }
 
-# # 处理合法电子邮件
-# print("\nProcessing legitimate email...")
-# legitimate_result = compiled_graph.invoke({
-#     "email": legitimate_email,
-#     "is_spam": None,
-#     "spam_reason": None,
-#     "email_category": None,
-#     "draft_response": None,
-#     "messages": []
-# })
-
-# 处理垃圾邮件
-# print("\nProcessing spam email...")
-# spam_result = compiled_graph.invoke({
-#     "email": spam_email,
-#     "is_spam": None,
-#     "spam_reason": None,
-#     "email_category": None,
-#     "draft_response": None,
-#     "messages": []
-# })
-
 def classify_email(email):
     result = compiled_graph.invoke({
         "email": email,
@@ -268,5 +246,5 @@ def classify_email(email):
     return f"is_spam:{result.get("is_spam")}\nspam_reason:{result.get("spam_reason")}"
 
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
+demo = gr.Interface(fn=classify_email, inputs="text", outputs="text")
 demo.launch()