Update app.py

app.py

@@ -45,7 +45,7 @@ logger = logging.getLogger()
 # ---------------------- Streamlit Page Configuration ---------------------------
 st.set_page_config(page_title="AI Clinical Intelligence Hub", layout="wide")
 
-# ---------------------- Initialize OpenAI Client ---------------------------
+# ---------------------- Initialize OpenAI SDK ---------------------------
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 PUB_EMAIL = os.getenv("PUB_EMAIL", "")
 
@@ -53,10 +53,8 @@ if not OPENAI_API_KEY:
     st.error("OpenAI API key must be set as an environment variable (OPENAI_API_KEY).")
     st.stop()
 
-# Instantiate the OpenAI client
-client = openai.OpenAI(
-    api_key=OPENAI_API_KEY
-)
+# Set the OpenAI API key
+openai.api_key = OPENAI_API_KEY
 
 # ---------------------- Load spaCy Model ---------------------------
 try:
@@ -494,7 +492,7 @@ class SimpleMedicalKnowledge(MedicalKnowledgeBase):
             """
 
             # Make the API request to OpenAI GPT-4
-            response = client.chat.completions.create(
+            response = openai.ChatCompletion.create(
                 model="gpt-4",
                 messages=[
                     {"role": "system", "content": "You are a helpful medical assistant."},
@@ -505,7 +503,7 @@ class SimpleMedicalKnowledge(MedicalKnowledgeBase):
             )
 
             # Extract the answer from the response
-            answer = response.choices[0].message.content.strip()
+            answer = response.choices[0].message['content'].strip()
 
             logger.info("Successfully retrieved data from OpenAI GPT-4.")