Update app.py

app.py

@@ -504,7 +504,7 @@ class SimpleMedicalKnowledge(MedicalKnowledgeBase):
                     {"role": "system", "content": "You are a helpful medical assistant."},
                     {"role": "user", "content": prompt}
                 ],
-                model="gpt-4",
+                model="gpt-4",  # Corrected model name
                 max_tokens=500,
                 temperature=0.7,
             )
@@ -781,16 +781,14 @@ def initialize_session_state():
         st.session_state.dashboard = Dashboard()
     if 'automated_reports' not in st.session_state:
         st.session_state.automated_reports = AutomatedReports()
+    if 'openai_client' not in st.session_state:
+        st.session_state.openai_client = st.session_state.get('openai_client', None)
     if 'diagnosis_support' not in st.session_state:
-        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()
+        st.session_state.diagnosis_support = SimpleDiagnosis(client=st.session_state.openai_client)
     if 'knowledge_base' not in st.session_state:
-        st.session_state.knowledge_base = SimpleMedicalKnowledge(nlp_model=nlp, client=st.session_state.get('openai_client'))
+        st.session_state.knowledge_base = SimpleMedicalKnowledge(nlp_model=nlp, client=st.session_state.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."""