Update app.py

app.py

@@ -16,6 +16,18 @@ import faiss
 # Initialize OpenAI client
 client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
 
+def get_embedding(text: str, model="text-embedding-ada-002"):
+    """從 OpenAI API 獲取文本嵌入"""
+    try:
+        response = client.embeddings.create(
+            input=[text],
+            model=model
+        )
+        return response.data[0]['embedding']
+    except Exception as e:
+        print(f"Error getting embedding: {e}")
+        return None
+        
 def initialize_embeddings_from_pkl(pkl_path: str, faiss_path: str):
     """從 PKL 檔案載入嵌入，並加載 FAISS 索引"""
     if not os.path.exists(pkl_path):