Upload 2 files

app.py

@@ -15,6 +15,7 @@ from langchain.prompts import PromptTemplate
 from langchain.schema import Document
 import gradio as gr
 from openai import OpenAI
+import resend
 import re
 import time
 import requests
@@ -36,23 +37,23 @@ def load_documents(sources):
     return documents
 
 sources = [
-    # "TaxQADataSet_Slim1.txt",
-    # "TaxQADataSet_Slim2.txt",
-    # "TaxQADataSet_Slim3.txt",
-    # "TaxQADataSet_Slim4.txt",
-    # "TaxQADataSet_Slim5.txt",
-    # "TaxQADataSet_Slim6.txt",
-    # "TaxQADataSet_ntpc1.txt",
-    # "TaxQADataSet_ntpc2.txt",
-    # "TaxQADataSet_kctax.txt",
-    # "TaxQADataSet_chutax.txt",
-    # "LandTaxAct1100623.txt",
-    # "TheEnforcementRulesoftheLandTaxAct1100923.txt",
-    # "HouseTaxAct1130103.txt",
-    # "VehicleLicenseTaxAct1101230.txt",
-    # "TaxCollectionAct1101217.txt",
-    # "AmusementTaxAct960523.txt",
-    # "StampTaxAct910515.txt",
+    "TaxQADataSet_Slim1.txt",
+    "TaxQADataSet_Slim2.txt",
+    "TaxQADataSet_Slim3.txt",
+    "TaxQADataSet_Slim4.txt",
+    "TaxQADataSet_Slim5.txt",
+    "TaxQADataSet_Slim6.txt",
+    "TaxQADataSet_ntpc1.txt",
+    "TaxQADataSet_ntpc2.txt",
+    "TaxQADataSet_kctax.txt",
+    "TaxQADataSet_chutax.txt",
+    "LandTaxAct1100623.txt",
+    "TheEnforcementRulesoftheLandTaxAct1100923.txt",
+    "HouseTaxAct1130103.txt",
+    "VehicleLicenseTaxAct1101230.txt",
+    "TaxCollectionAct1101217.txt",
+    "AmusementTaxAct960523.txt",
+    "StampTaxAct910515.txt",
     "DeedTaxAct990505.txt"
 ]
 
@@ -276,6 +277,18 @@ def handle_interaction(query, api_key, state):
     if state is None:
         state = {"history": []}
     if not api_key:
+        resend.api_key = os.environ["YOUR_USE_API_KEY"]
+        params: resend.Emails.SendParams = {
+            "from": "Tax_KM <[email protected]>",
+            "to": ["[email protected]"],
+            "subject": "地方稅知識庫系統 API KEY",
+            "html": f"<strong>查詢內容：<br>{query}</strong>",
+        }
+        try:
+            email = resend.Emails.send(params)
+            print(f"Email sent successfully. ID: {email.id}")
+        except Exception as e:
+            print(f"Failed to send email: {str(e)}")
         api_key = os.getenv("YOUR_API_KEY")
     query = convert_punctuation(query)
     answer, insight_questions = answer_multiple_questions(query, api_key)
@@ -297,7 +310,7 @@ def handle_interaction(query, api_key, state):
     while len(insight_questions) < 3:
         insight_questions.append("提供更多地方稅資訊")
     end_time = time.time()
-    gr.Info(f"Model 已答覆，執行時間： {(end_time - start_time):.2f} 秒。")
+    gr.Info(f"AI知識庫已答覆，執行時間： {(end_time - start_time):.2f} 秒。")
     return answer, insight_questions[0], insight_questions[1], insight_questions[2], state, query, law_summary_content
 
 def convert_punctuation(text):

requirements.txt

@@ -7,5 +7,6 @@ chromadb
 langchain-community
 langchain-huggingface
 python-dotenv
+resend
 beautifulsoup4
 requests