Update app.py

app.py

@@ -25,7 +25,7 @@ from Bio import Entrez  # Ensure BioPython is installed
 
 from dotenv import load_dotenv
 import requests
-import openai  # Added for OpenAI GPT-4 integration
+import openai  # Updated for OpenAI SDK v1.0
 import ast
 import logging
 
@@ -42,7 +42,6 @@ logging.basicConfig(
 logger = logging.getLogger()
 
 # ---------------------- Streamlit Page Configuration ---------------------------
-# This must be the first Streamlit command in the script
 st.set_page_config(page_title="AI Clinical Intelligence Hub", layout="wide")
 
 # ---------------------- Initialize External Clients ---------------------------
@@ -54,7 +53,7 @@ if not OPENAI_API_KEY:
     st.error("OpenAI API key must be set as an environment variable (OPENAI_API_KEY).")
     st.stop()
 
-openai.api_key = OPENAI_API_KEY
+openai_client = openai.OpenAI(api_key=OPENAI_API_KEY)  # Instantiate OpenAI client
 
 # Load spaCy model with error handling
 try:
@@ -390,7 +389,7 @@ class DiagnosisSupport(ABC):
 class SimpleDiagnosis(DiagnosisSupport):
     """Provides a simple diagnosis example, based on the Logistic regression model."""
     def __init__(self):
-        self.model: LogisticRegressionTrainer = LogisticRegressionTrainer()
+        self.model_trainer: LogisticRegressionTrainer = LogisticRegressionTrainer()
 
     def diagnose(
         self, 
@@ -401,7 +400,7 @@ class SimpleDiagnosis(DiagnosisSupport):
         **kwargs
     ) -> pd.DataFrame:
         try:
-            result = self.model.invoke(data, target_col=target_col, columns=columns)
+            result = self.model_trainer.invoke(data, target_col=target_col, columns=columns)
             if "accuracy" in result:
                 return pd.DataFrame({
                     diagnosis_key: [f"Model Accuracy: {result['accuracy']:.2%}"],
@@ -466,8 +465,9 @@ class MedicalKnowledgeBase(ABC):
 
 class SimpleMedicalKnowledge(MedicalKnowledgeBase):
     """Enhanced Medical Knowledge Class using OpenAI GPT-4."""
-    def __init__(self):
-        self.nlp = nlp  # Using the loaded spaCy model
+    def __init__(self, client: openai.OpenAI, nlp_model):
+        self.nlp = nlp_model  # Using the loaded spaCy model
+        self.client = client
 
     def search_medical_info(self, query: str, pub_email: str = "") -> str:
         """
@@ -492,20 +492,18 @@ class SimpleMedicalKnowledge(MedicalKnowledgeBase):
             """
 
             # Make the API request to OpenAI GPT-4
-            response = openai.ChatCompletion.create(
+            response = self.client.chat_completions.create(
                 model="gpt-4",
                 messages=[
                     {"role": "system", "content": "You are a helpful medical assistant."},
                     {"role": "user", "content": prompt}
                 ],
                 max_tokens=500,
-                n=1,
-                stop=None,
                 temperature=0.7,
             )
 
             # 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.")
 
@@ -775,7 +773,7 @@ def initialize_session_state():
     if 'treatment_recommendation' not in st.session_state:
         st.session_state.treatment_recommendation = BasicTreatmentRecommendation()
     if 'knowledge_base' not in st.session_state:
-        st.session_state.knowledge_base = SimpleMedicalKnowledge()
+        st.session_state.knowledge_base = SimpleMedicalKnowledge(openai_client, nlp)
     if 'pub_email' not in st.session_state:
         st.session_state.pub_email = PUB_EMAIL  # Load PUB_EMAIL from environment variables