Upload 10 files

app.py

@@ -93,6 +93,34 @@ except:
     st.session_state.messages = []
     st.session_state.gemini_history = []
     print('new_cache made')
+
+# Inicializar el modelo con system prompt solo si no existe o si cambiamos de chat
+if not hasattr(st.session_state, 'puv_model') or st.session_state.get('current_chat_id') != st.session_state.chat_id:
+    system_prompt = get_puv_system_prompt()
+    st.session_state.puv_model = genai.GenerativeModel(
+        model_name='gemini-2.0-flash',
+        system_instruction=system_prompt
+    )
+    st.session_state.current_chat_id = st.session_state.chat_id
+    
+    # Inicializar el chat con el historial existente
+    history_for_model = []
+    for msg in st.session_state.gemini_history:
+        if isinstance(msg, dict) and 'role' in msg and 'parts' in msg:
+            # El formato ya es correcto
+            history_for_model.append(msg)
+        elif isinstance(msg, dict) and 'role' in msg and 'content' in msg:
+            # Convertir del formato antiguo al nuevo
+            history_for_model.append({
+                'role': msg['role'],
+                'parts': [{'text': msg['content']}]
+            })
+    
+    st.session_state.puv_chat = st.session_state.puv_model.start_chat(
+        history=history_for_model
+    )
+
+# Modelo estándar para otras operaciones (como generar títulos)
 st.session_state.model = genai.GenerativeModel('gemini-2.0-flash')
 st.session_state.chat = st.session_state.model.start_chat(
     history=st.session_state.gemini_history,
@@ -152,17 +180,8 @@ if prompt := st.chat_input('¿En qué puedo ayudarte con tu Propuesta Única de
         )
     )
     
-    # Obtener el system prompt iterativo
-    system_prompt = get_puv_system_prompt()
-    
-    # Crear una instancia del modelo con el system prompt
-    model = genai.GenerativeModel(
-        model_name='gemini-2.0-flash',
-        system_instruction=system_prompt
-    )
-    
-    # Enviar mensaje al AI con system prompt
-    response_stream = model.generate_content(
+    # Usar el chat con system prompt para generar la respuesta
+    response = st.session_state.puv_chat.send_message(
         prompt,
         stream=True,
     )
@@ -207,19 +226,8 @@ if prompt := st.chat_input('¿En qué puedo ayudarte con tu Propuesta Única de
         )
     )
     
-    # Actualizar el historial de Gemini (adaptado para la nueva API)
-    if not hasattr(st.session_state, 'gemini_history'):
-        st.session_state.gemini_history = []
-    
-    st.session_state.gemini_history.append({
-        'role': 'user',
-        'parts': [{'text': prompt}]
-    })
-    
-    st.session_state.gemini_history.append({
-        'role': 'model',
-        'parts': [{'text': full_response}]
-    })
+    # Actualizar el historial de Gemini
+    st.session_state.gemini_history = st.session_state.puv_chat.history
     
     # Save to file
     joblib.dump(