Update app.py

app.py

@@ -28,7 +28,7 @@ from Bio import Entrez  # Ensure BioPython is installed
 from dotenv import load_dotenv
 import requests
 import openai  # Updated for OpenAI SDK v1.0.0+
-from openai import OpenAIError, RateLimitError, BadRequestError
+from openai import OpenAIError, RateLimitError, BadRequestError, OpenAI
 
 # ---------------------- Load Environment Variables ---------------------------
 load_dotenv()
@@ -53,13 +53,13 @@ if not OPENAI_API_KEY:
     st.error("OpenAI API key must be set as an environment variable (OPENAI_API_KEY).")
     st.stop()
 
-# Set the OpenAI API key
-openai.api_key = OPENAI_API_KEY
-
-# Update OpenAI base_url and proxies if needed
-# Example:
-# openai.api_base = "https://api.openai.com/v1"
-# openai.proxies = {"https": "http://proxy.example.com:8080"}
+# Instantiate the OpenAI client
+try:
+    client = OpenAI(api_key=OPENAI_API_KEY)
+except Exception as e:
+    st.error(f"Failed to initialize OpenAI client: {e}")
+    logger.error(f"Failed to initialize OpenAI client: {e}")
+    st.stop()
 
 # ---------------------- Load spaCy Model ---------------------------
 try:
@@ -394,8 +394,9 @@ class DiagnosisSupport(ABC):
 
 class SimpleDiagnosis(DiagnosisSupport):
     """Provides a simple diagnosis example, based on the Logistic regression model."""
-    def __init__(self):
+    def __init__(self, client: OpenAI):
         self.model_trainer: LogisticRegressionTrainer = LogisticRegressionTrainer()
+        self.client = client  # Using the OpenAI client
 
     def diagnose(
         self, 
@@ -471,8 +472,9 @@ class MedicalKnowledgeBase(ABC):
 
 class SimpleMedicalKnowledge(MedicalKnowledgeBase):
     """Enhanced Medical Knowledge Class using OpenAI GPT-4."""
-    def __init__(self, nlp_model):
+    def __init__(self, nlp_model, client: OpenAI):
         self.nlp = nlp_model  # Using the loaded spaCy model
+        self.client = client  # Using the OpenAI client
 
     def search_medical_info(self, query: str, pub_email: str = "") -> str:
         """
@@ -497,12 +499,12 @@ class SimpleMedicalKnowledge(MedicalKnowledgeBase):
             """
 
             # Make the API request to OpenAI GPT-4
-            response = openai.ChatCompletion.create(
-                model="gpt-4",
+            response = self.client.chat.completions.create(
                 messages=[
                     {"role": "system", "content": "You are a helpful medical assistant."},
                     {"role": "user", "content": prompt}
                 ],
+                model="gpt-4",
                 max_tokens=500,
                 temperature=0.7,
             )
@@ -780,13 +782,15 @@ def initialize_session_state():
     if 'automated_reports' not in st.session_state:
         st.session_state.automated_reports = AutomatedReports()
     if 'diagnosis_support' not in st.session_state:
-        st.session_state.diagnosis_support = SimpleDiagnosis()
+        st.session_state.diagnosis_support = SimpleDiagnosis(client=st.session_state.get('openai_client'))
     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(nlp_model=nlp)
+        st.session_state.knowledge_base = SimpleMedicalKnowledge(nlp_model=nlp, client=st.session_state.get('openai_client'))
     if 'pub_email' not in st.session_state:
         st.session_state.pub_email = PUB_EMAIL  # Load PUB_EMAIL from environment variables
+    if 'openai_client' not in st.session_state:
+        st.session_state.openai_client = client  # Store the OpenAI client
 
 def data_management_section():
     """Handles the data management section in the sidebar."""