Update app.py

app.py

@@ -52,11 +52,11 @@ def flatten_json_obj(obj, parent_key="", sep="."):
         items[parent_key] = obj
     return items
 
-# --- Embedding function ---
+# --- Embedding function (updated for openai>=1.0.0) ---
 def get_embedding(text):
-    openai.api_key = OPENAI_API_KEY
-    resp = openai.Embedding.create(input=text, model=EMBEDDING_MODEL)
-    return resp['data'][0]['embedding']
+    client = openai.OpenAI(api_key=OPENAI_API_KEY)
+    response = client.embeddings.create(input=[text], model=EMBEDDING_MODEL)
+    return response.data[0].embedding
 
 # --- Ensure DB Table (accumulates all uploads, never deletes old data) ---
 def ensure_table():
@@ -154,7 +154,7 @@ class SQLiteVectorRetriever(BaseRetriever):
     def _get_relevant_documents(self, query, run_manager=None, **kwargs):
         return query_vector_db(query, self.top_k)
 
-llm = OpenAI(model="gpt-4.1", openai_api_key=OPENAI_API_KEY, temperature=0)
+llm = OpenAI(model="gpt-4o", openai_api_key=OPENAI_API_KEY, temperature=0)
 retriever = SQLiteVectorRetriever(top_k=5)
 qa_chain = RetrievalQA.from_chain_type(
     llm=llm,