Update app.py

app.py

@@ -11,8 +11,6 @@ import chardet
 import pandas as pd
 import plotly.graph_objs as go
 
-os.environ["GROQ_API_KEY"] = "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"
-
 try:
     SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
 except NameError:
@@ -68,10 +66,10 @@ def setup(file_path):
 
     return retriever, chain_type_kwargs
 
-def setup_qa(file_path, model_name):
+def setup_qa(file_path, model_name, api_key):
     retriever, chain_type_kwargs = setup(file_path)
     llm = ChatGroq(
-        groq_api_key=os.environ["GROQ_API_KEY"],
+        groq_api_key=api_key,
         model_name=model_name,
     )
     qa = RetrievalQA.from_chain_type(
@@ -83,20 +81,23 @@ def setup_qa(file_path, model_name):
     )
     return qa
 
-def chat_with_models(file, model_a, model_b, history_a, history_b, question):
+def chat_with_models(file, model_a, model_b, api_key, history_a, history_b, question):
     if file is None:
-        return history_a + [("請上傳文件。", None)], history_b + [("請上傳文件。", None)]
+        return history_a + [("請上傳文件。", None)], history_b + [("請上傳文件。", None)], ""
+    
+    if not api_key:
+        return history_a + [("請輸入 API 金鑰。", None)], history_b + [("請輸入 API 金鑰。", None)], ""
     
     file_path = file.name
     _, extension = os.path.splitext(file_path)
 
     if extension.lower() not in ['.pdf', '.txt']:
         error_message = "只能上傳PDF或TXT檔案，不接受其他的格式。"
-        return history_a + [(error_message, None)], history_b + [(error_message, None)]
+        return history_a + [(error_message, None)], history_b + [(error_message, None)], ""
 
     try:
-        qa_a = setup_qa(file_path, model_a)
-        qa_b = setup_qa(file_path, model_b)
+        qa_a = setup_qa(file_path, model_a, api_key)
+        qa_b = setup_qa(file_path, model_b, api_key)
 
         response_a = qa_a.invoke(question)
         response_b = qa_b.invoke(question)
@@ -104,10 +105,10 @@ def chat_with_models(file, model_a, model_b, history_a, history_b, question):
         history_a.append((question, response_a["result"]))
         history_b.append((question, response_b["result"]))
 
-        return history_a, history_b
+        return history_a, history_b, ""
     except Exception as e:
         error_message = f"遇到錯誤：{str(e)}"
-        return history_a + [(error_message, None)], history_b + [(error_message, None)]
+        return history_a + [(error_message, None)], history_b + [(error_message, None)], ""
 
 def load_or_create_df():
     if os.path.exists(CSV_PATH):
@@ -154,6 +155,7 @@ def create_demo():
             model_a = gr.Dropdown(choices=models, label="Model A", value=models[0])
             model_b = gr.Dropdown(choices=models, label="Model B", value=models[-1])
         
+        api_key = gr.Textbox(label="Enter your Groq API Key", type="password")
         file_input = gr.File(label="Upload PDF or TXT file")
         
         with gr.Row():
@@ -176,8 +178,8 @@ def create_demo():
         
         send_btn.click(
             fn=chat_with_models,
-            inputs=[file_input, model_a, model_b, chat_a, chat_b, question],
-            outputs=[chat_a, chat_b],
+            inputs=[file_input, model_a, model_b, api_key, chat_a, chat_b, question],
+            outputs=[chat_a, chat_b, question],
         )
         
         def create_evaluate_fn(eval_type):