Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import os
-api_token = os.getenv("HF_TOKEN")
+#api_token = os.getenv("HF_TOKEN")
 
 
 from langchain_community.vectorstores import FAISS
@@ -15,6 +15,15 @@ from langchain.memory import ConversationBufferMemory
 from langchain_community.llms import HuggingFaceEndpoint
 import torch
 
+
+# 獲取 OpenAI API 密鑰（初始不使用固定密鑰）
+api_key_env = os.getenv("OPENAI_API")
+if api_key_env:
+    openai.api_key = api_key_env
+    logger.info("OpenAI API 密鑰已設置。")
+else:
+    logger.info("未設置固定的 OpenAI API 密鑰。將使用使用者提供的密鑰。")
+    
 list_llm = ["meta-llama/Meta-Llama-3-8B-Instruct", "mistralai/Mistral-7B-Instruct-v0.2"]  
 list_llm_simple = [os.path.basename(llm) for llm in list_llm]
 
@@ -46,14 +55,14 @@ def initialize_llmchain(llm_model, temperature, max_tokens, top_k, vector_db, pr
     if llm_model == "meta-llama/Meta-Llama-3-8B-Instruct":
         llm = HuggingFaceEndpoint(
             repo_id=llm_model,
-            huggingfacehub_api_token = api_token,
+            huggingfacehub_api_token = api_key_env,
             temperature = temperature,
             max_new_tokens = max_tokens,
             top_k = top_k,
         )
     else:
         llm = HuggingFaceEndpoint(
-            huggingfacehub_api_token = api_token,
+            huggingfacehub_api_token = api_key_env,
             repo_id=llm_model, 
             temperature = temperature,
             max_new_tokens = max_tokens,