Create session_state.py

src/session_state.py

@@ -0,0 +1,46 @@
+"""
+This module handles session state initialization and management.
+"""
+
+import streamlit as st
+import uuid
+
+
+def initialize_session_state():
+    """
+    Initialize all session state variables needed for the application.
+    """
+    # Main conversation history
+    if 'history' not in st.session_state:
+        st.session_state.history = []
+        
+    # Unique consultation ID for tracking sessions
+    if 'consultation_id' not in st.session_state:
+        st.session_state.consultation_id = str(uuid.uuid4())[:8]
+        
+    # RAG feature toggle
+    if 'use_rag' not in st.session_state:
+        st.session_state.use_rag = True
+        
+    # Processing state for showing typing indicator
+    if 'processing' not in st.session_state:
+        st.session_state.processing = False
+        
+    # Report generation state
+    if 'show_report_form' not in st.session_state:
+        st.session_state.show_report_form = False
+        
+    if 'report_step' not in st.session_state:
+        st.session_state.report_step = 0
+        
+    # Patient information for reports
+    if 'patient_info' not in st.session_state:
+        st.session_state.patient_info = {"name": "", "age": "", "gender": ""}
+        
+    # PDF report data
+    if 'pdf_data' not in st.session_state:
+        st.session_state.pdf_data = None
+        
+    # Email form visibility
+    if 'show_email_form' not in st.session_state:
+        st.session_state.show_email_form = False