Update session_state.py

session_state.py

@@ -1,11 +1,10 @@
 import streamlit as st
 import joblib
 import os
-import uuid
 from google import genai
 from google.genai import types
 
-DEFAULT_GEMINI_MODEL = 'gemini-2.5-flash-lite'
+DEFAULT_GEMINI_MODEL = 'gemini-3.1-flash-lite-preview'
 DATA_DIR = 'data'
 PAST_CHATS_LIST_PATH = f'{DATA_DIR}/past_chats_list'
 
@@ -43,12 +42,6 @@ class SessionState:
 
         if 'system_instruction' not in st.session_state:
             st.session_state.system_instruction = None
-
-        if 'user_id' not in st.session_state:
-            st.session_state.user_id = f'local-{uuid.uuid4()}'
-
-        if 'session_store' not in st.session_state:
-            st.session_state.session_store = None
     
     # Getters y setters para cada propiedad
     @property
@@ -123,22 +116,6 @@ class SessionState:
     def system_instruction(self, value):
         st.session_state.system_instruction = value
 
-    @property
-    def user_id(self):
-        return st.session_state.user_id
-
-    @user_id.setter
-    def user_id(self, value):
-        st.session_state.user_id = value
-
-    @property
-    def session_store(self):
-        return st.session_state.session_store
-
-    @session_store.setter
-    def session_store(self, value):
-        st.session_state.session_store = value
-    
     # Métodos de utilidad
     def add_message(self, role, content, avatar=None):
         """Añade un mensaje al historial"""
@@ -154,11 +131,6 @@ class SessionState:
         """Limpia el prompt del estado de la sesión"""
         self.prompt = None
 
-    def set_storage(self, user_id, session_store=None):
-        """Configura el usuario actual y el store opcional (Firebase)."""
-        self.user_id = user_id
-        self.session_store = session_store
-    
     def initialize_model(self, model_name=None, api_key=None):
         """Inicializa el modelo de IA"""
         if model_name is None:
@@ -242,13 +214,10 @@ class SessionState:
         if chat_id is None:
             chat_id = self.chat_id
         
-        if self.session_store:
-            self.session_store.save_chat_history(self.user_id, chat_id, self.messages, self.gemini_history)
-            return
-
-        os.makedirs(self._user_data_dir(), exist_ok=True)
+        serialized_history = self._serialize_gemini_history(self.gemini_history)
+        os.makedirs(DATA_DIR, exist_ok=True)
         joblib.dump(self.messages, self._st_messages_path(chat_id))
-        joblib.dump(self.gemini_history, self._gemini_messages_path(chat_id))
+        joblib.dump(serialized_history, self._gemini_messages_path(chat_id))
     
     def load_chat_history(self, chat_id=None):
         """Carga el historial del chat"""
@@ -256,18 +225,9 @@ class SessionState:
             chat_id = self.chat_id
         
         try:
-            if self.session_store:
-                messages, gemini_history = self.session_store.load_chat_history(self.user_id, chat_id)
-                if messages is not None and gemini_history is not None:
-                    self.messages = messages
-                    self.gemini_history = gemini_history
-                    return True
-                self.messages = []
-                self.gemini_history = []
-                return False
-
             self.messages = joblib.load(self._st_messages_path(chat_id))
-            self.gemini_history = joblib.load(self._gemini_messages_path(chat_id))
+            loaded_history = joblib.load(self._gemini_messages_path(chat_id))
+            self.gemini_history = self._deserialize_gemini_history(loaded_history)
             return True
         except (FileNotFoundError, EOFError):
             self.messages = []
@@ -275,13 +235,45 @@ class SessionState:
             return False
 
     def _st_messages_path(self, chat_id):
-        return f'{self._user_data_dir()}/{chat_id}-st_messages'
+        return f'{DATA_DIR}/{chat_id}-st_messages'
 
     def _gemini_messages_path(self, chat_id):
-        return f'{self._user_data_dir()}/{chat_id}-gemini_messages'
+        return f'{DATA_DIR}/{chat_id}-gemini_messages'
+
+    def _serialize_gemini_history(self, history):
+        """Convierte tipos del SDK (Content/Part) a diccionarios serializables."""
+        serialized = []
+        for item in history or []:
+            if isinstance(item, dict):
+                serialized.append(item)
+                continue
+            if hasattr(item, "model_dump"):
+                serialized.append(item.model_dump(mode="python"))
+                continue
+            if hasattr(item, "to_dict"):
+                serialized.append(item.to_dict())
+                continue
+            serialized.append(item)
+        return serialized
 
-    def _user_data_dir(self):
-        return f'{DATA_DIR}/users/{self.user_id}'
+    def _deserialize_gemini_history(self, history):
+        """Reconstruye Content para rehidratar chat history en google-genai."""
+        deserialized = []
+        for item in history or []:
+            if isinstance(item, dict) and "role" in item and "parts" in item:
+                role = item.get("role")
+                parts_data = item.get("parts", [])
+                parts = []
+                for part in parts_data:
+                    if isinstance(part, dict) and "text" in part:
+                        parts.append(types.Part(text=part["text"]))
+                    elif isinstance(part, str):
+                        parts.append(types.Part(text=part))
+                if parts:
+                    deserialized.append(types.Content(role=role, parts=parts))
+                    continue
+            deserialized.append(item)
+        return deserialized
     
     def has_messages(self):
         """Verifica si hay mensajes en el historial"""