Update app.py

app.py

@@ -56,7 +56,8 @@ if not OPENAI_API_KEY:
     st.error("OpenAI API key must be set as an environment variable (OPENAI_API_KEY).")
     st.stop()
 
-openai_client = openai.OpenAI(api_key=OPENAI_API_KEY)  # Instantiate OpenAI client
+# Set the OpenAI API key directly
+openai.api_key = OPENAI_API_KEY
 
 # Load spaCy model with error handling
 try:
@@ -468,9 +469,8 @@ class MedicalKnowledgeBase(ABC):
 
 class SimpleMedicalKnowledge(MedicalKnowledgeBase):
     """Enhanced Medical Knowledge Class using OpenAI GPT-4."""
-    def __init__(self, client: openai.OpenAI, nlp_model):
+    def __init__(self, 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:
         """
@@ -495,7 +495,7 @@ class SimpleMedicalKnowledge(MedicalKnowledgeBase):
             """
 
             # Make the API request to OpenAI GPT-4
-            response = self.client.chat.completions.create(
+            response = openai.ChatCompletion.create(
                 model="gpt-4",
                 messages=[
                     {"role": "system", "content": "You are a helpful medical assistant."},
@@ -776,7 +776,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(openai_client, nlp)
+        st.session_state.knowledge_base = SimpleMedicalKnowledge(nlp_model=nlp)
     if 'pub_email' not in st.session_state:
         st.session_state.pub_email = PUB_EMAIL  # Load PUB_EMAIL from environment variables