Upload 10 files

app.py

@@ -32,9 +32,6 @@ 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
@@ -158,29 +155,16 @@ 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()
     
-    # Preparar el contenido para la API de Gemini
-    contents = [
-        types.Content(
-            role="user",
-            parts=[
-                types.Part.from_text(text=prompt),
-            ],
-        ),
-    ]
-    
-    # Configurar el system prompt
-    generate_content_config = types.GenerateContentConfig(
-        response_mime_type="text/plain",
-        system_instruction=[
-            types.Part.from_text(text=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 = client.models.generate_content_stream(
-        model="gemini-2.0-flash",
-        contents=contents,
-        config=generate_content_config,
+    response_stream = model.generate_content(
+        prompt,
+        stream=True,
     )
     
     # Display assistant response in chat message container