Update app.py

app.py

@@ -55,7 +55,7 @@ if not OPENAI_API_KEY:
 
 # Instantiate the OpenAI client
 try:
-    client = OpenAI(api_key=OPENAI_API_KEY)
+   client = OpenAI(api_key=OPENAI_API_KEY) # Instantiating the client right here
 except Exception as e:
     st.error(f"Failed to initialize OpenAI client: {e}")
     logger.error(f"Failed to initialize OpenAI client: {e}")
@@ -763,6 +763,11 @@ def main():
 
 def initialize_session_state():
     """Initialize necessary components in Streamlit's session state."""
+
+    if 'openai_client' not in st.session_state:
+        # Instantiate the OpenAI client only if it doesn't exist in session state
+        st.session_state.openai_client = client # The one created earlier
+
     if 'data' not in st.session_state:
         st.session_state.data = {}  # Store pd.DataFrame under a name
     if 'data_ingestion' not in st.session_state:
@@ -781,14 +786,15 @@ 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.openai_client)
     if 'knowledge_base' not in st.session_state:
         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
+                st.session_state.pub_email = PUB_EMAIL  # Load PUB_EMAIL from environment variables
+    if 'treatment_recommendation' not in st.session_state:
+        st.session_state.treatment_recommendation = BasicTreatmentRecommendation()
 
 def data_management_section():
     """Handles the data management section in the sidebar."""
@@ -1169,4 +1175,4 @@ def medical_knowledge_section():
             st.error("Please enter a medical question to search.")
 
 if __name__ == "__main__":
-    main()
+    main()