Upload 10 files

app.py

@@ -2,9 +2,9 @@ import time
 import os
 import joblib
 import streamlit as st
-import google.generativeai as genai
+from google import genai
+from google.genai import types
 from dotenv import load_dotenv
-from google.generativeai import types
 from system_prompts import get_puv_system_prompt  # Importar el prompt iterativo
 
 # Función para cargar CSS personalizado
@@ -33,6 +33,9 @@ load_dotenv()
 GOOGLE_API_KEY=os.environ.get('GOOGLE_API_KEY')
 genai.configure(api_key=GOOGLE_API_KEY)
 
+# Inicializar el cliente de Gemini
+client = genai.Client(api_key=GOOGLE_API_KEY)
+
 new_chat_id = f'{time.time()}'
 MODEL_ROLE = 'ai'
 AI_AVATAR_ICON = '🤖'  # Cambia el emoji por uno de robot para coincidir con tu logo
@@ -156,27 +159,31 @@ 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 nueva instancia del modelo con el system prompt
-    model_with_system_prompt = genai.GenerativeModel(
-        model_name='gemini-2.0-flash',
-        system_instruction=system_prompt
-    )
+    # Preparar el contenido para la API de Gemini
+    contents = [
+        types.Content(
+            role="user",
+            parts=[
+                types.Part.from_text(text=prompt),
+            ],
+        ),
+    ]
     
-    # Crear un chat con el modelo que tiene el system prompt
-    chat_with_system = model_with_system_prompt.start_chat(history=[])
+    # Configurar el system prompt
+    generate_content_config = types.GenerateContentConfig(
+        response_mime_type="text/plain",
+        system_instruction=[
+            types.Part.from_text(text=system_prompt),
+        ],
+    )
     
     # Enviar mensaje al AI con system prompt
-    response = chat_with_system.send_message(
-        prompt,
-        stream=True,
+    response_stream = client.models.generate_content_stream(
+        model="gemini-2.0-flash",
+        contents=contents,
+        config=generate_content_config,
     )
     
-    # Eliminar la línea duplicada que usa st.session_state.chat
-    # response = st.session_state.chat.send_message(
-    #    prompt,
-    #    stream=True,
-    # )
-    
     # Display assistant response in chat message container
     with st.chat_message(
         name=MODEL_ROLE,
@@ -184,24 +191,25 @@ if prompt := st.chat_input('¿En qué puedo ayudarte con tu Propuesta Única de
     ):
         message_placeholder = st.empty()
         full_response = ''
-        assistant_response = response
         
         # Añade un indicador de "escribiendo..."
         typing_indicator = st.empty()
         typing_indicator.markdown("*RoboCopy está escribiendo...*")
         
         # Streams in a chunk at a time
-        for chunk in response:
-            # Simulate stream of chunk
-            words = chunk.text.split(' ')
-            for i, word in enumerate(words):
-                full_response += word
-                # Añadir espacio solo si no es la última palabra del chunk
-                if i < len(words) - 1:
-                    full_response += ' '
-                time.sleep(0.1)  # Velocidad ajustada para mejor legibilidad
-                # Rewrites with a cursor at end
-                message_placeholder.write(full_response + '▌')
+        for chunk in response_stream:
+            if hasattr(chunk, 'text'):
+                # Simulate stream of chunk
+                words = chunk.text.split(' ')
+                for i, word in enumerate(words):
+                    full_response += word
+                    # Añadir espacio solo si no es la última palabra del chunk
+                    if i < len(words) - 1:
+                        full_response += ' '
+                    time.sleep(0.1)  # Velocidad ajustada para mejor legibilidad
+                    # Rewrites with a cursor at end
+                    message_placeholder.write(full_response + '▌')
+        
         # Elimina el indicador de escritura
         typing_indicator.empty()
         # Write full message with placeholder
@@ -211,11 +219,25 @@ if prompt := st.chat_input('¿En qué puedo ayudarte con tu Propuesta Única de
     st.session_state.messages.append(
         dict(
             role=MODEL_ROLE,
-            content=st.session_state.chat.history[-1].parts[0].text,
+            content=full_response,
             avatar=AI_AVATAR_ICON,
         )
     )
-    st.session_state.gemini_history = st.session_state.chat.history
+    
+    # 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}]
+    })
+    
     # Save to file
     joblib.dump(
         st.session_state.messages,