Update app.py

app.py

@@ -1,11 +1,11 @@
 import os
 import json
-import openai
 from fastapi import FastAPI, Request
+from openai import OpenAI
 
 app = FastAPI()
 
-# Load knowledge base
+# Load kb.json
 with open("kb.json") as f:
     kb = json.load(f)
 
@@ -17,8 +17,8 @@ system_prompt += (
     "'I'm not sure about that. Let me connect you with a human agent.'"
 )
 
-openai.api_key = os.getenv("OPENAI_API_KEY")
-openai.api_base = os.getenv("OPENAI_API_BASE", "https://fast.typegpt.net/v1")
+# Initialize OpenAI client
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"), api_base=os.getenv("OPENAI_API_BASE", "https://fast.typegpt.net/v1"))
 
 @app.post("/ask")
 async def ask(request: Request):
@@ -34,7 +34,7 @@ async def ask(request: Request):
     ]
 
     try:
-        response = openai.ChatCompletion.create(
+        response = client.chat.completions.create(
             model="gpt-4o-mini",
             messages=messages,
             temperature=0.0,
@@ -42,7 +42,7 @@ async def ask(request: Request):
         )
         answer = response.choices[0].message.content.strip()
     except Exception as e:
-        print(f"Error calling OpenAI: {e}")
+        print(f"OpenAI API error: {e}")
         answer = "Sorry, I'm having trouble answering right now."
 
     return {"reply": answer}