Upload 15 files

app.py

@@ -1,14 +1,13 @@
 from frontend_pages import st
 from frontend_pages.ConfigPage import ConfigPage
 from frontend_pages.ChatModelPage import ChatModelPage
-from frontend_pages.Dashboard import Dashboard
 
 
-PAGES = {
-    "Configurazione": ConfigPage,
-    "Dashboard": ChatModelPage,
-}
+# PAGES = {
+#     "Dashboard": ChatModelPage,
+#     # "Configurazione": ConfigPage,
+# }
 
 if __name__ == "__main__":
-    PAGES[st.sidebar.selectbox("Select a page", list(PAGES.keys()))]().render()
+    ChatModelPage().render()
     # PAGES["Chat with AI Model"]().render()

frontend_pages/ChatModelPage.py

@@ -1,4 +1,5 @@
 from . import os, APIS, st, re, json, requests, pd, PromptRequest, TranscriptRequest, UpdateTasksFromTranscriptRequest, explain_agents, stylable_container
+from APIModules.ObjectModels.Task import Task
 from .BasePage import BasePage
 
 from .modals import (
@@ -121,6 +122,20 @@ class ChatModelPage(BasePage):
         super().__init__()
         self.model = None
         self.router_agent = RouterAgent()
+        with st.sidebar:
+            initial_state_from_json : dict = st.session_state.get("INITIAL_STATE", {})
+            if st.button("use_preloaded_state"):
+                with open("initial_state.json") as f:
+                    initial_state_from_json : dict = json.load(f)
+                    st.session_state.INITIAL_STATE = initial_state_from_json
+                    st.rerun()
+            initial_team_members = st.session_state.get('team')
+            if not initial_team_members:st.session_state.team = initial_state_from_json.get('team', ["Alessio", "Joy", "Nicola C", "Nicola D", "Dragosh"])
+            initial_tasks = st.session_state.get('tasks')
+            if not initial_tasks:st.session_state.tasks = [Task(**task).model_dump() for task in initial_state_from_json.get('tasks', [])]
+            initial_assignments = st.session_state.get('assignments')
+            if not initial_tasks:st.session_state.assignments = initial_state_from_json.get('assignments', {})
+
 
     def render(self):
         st.title("Dashboard")

frontend_pages/Dashboard.py

@@ -4,6 +4,8 @@ from .BasePage import BasePage
 from .modals import get_base64_image, create_call_modal, upload_call_modal_and_actions
 
 class Dashboard(BasePage):
+
+    
     def render(self):
         st.title("Dashboard")
         self.sidebar_elements()