Update app.py

app.py

@@ -1,57 +1,28 @@
+
+
 import streamlit as st
 import os
-import google.generativeai as genai
+from google.generativeai import configure, GenerativeModel
 from dotenv import load_dotenv
-import fitz
+import fitz  # Para leer PDF si lo usarás más adelante
 import base64
-from io import StringIO
-from google import genai
-from google.genai import types
 
+# Cargar las variables del entorno
 load_dotenv()
+GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
 
-GEMINI_API_KEY = os.environ["GEMINI_API_KEY"]
-
-
-def generate():
-    client = genai.Client(
-        api_key=os.environ.get("GEMINI_API_KEY"),
-    )
-
-    model = "gemini-2.5-pro-exp-03-25"
-    contents = [
-        types.Content(
-            role="user",
-            parts=[
-                types.Part.from_text(text="""INSERT_INPUT_HERE"""),
-            ],
-        ),
-    ]
-    generate_content_config = types.GenerateContentConfig(
-        temperature=1,
-        top_p=0.95,
-        top_k=64,
-        max_output_tokens=65536,
-        response_mime_type="text/plain",
-    )
-
-    for chunk in client.models.generate_content_stream(
-        model=model,
-        contents=contents,
-        config=generate_content_config,
-    ):
-        print(chunk.text, end="")
-
-if __name__ == "__main__":
-    generate()
-
+# Configurar Gemini API
+configure(api_key=GEMINI_API_KEY)
 
+# Crear el modelo
+model = GenerativeModel('gemini-pro')
 
 st.title("Gemini chatbot")
 st.subheader("Hola, hazle una pregunta a Gemini :)")
 
+# Input del usuario
 prompt = st.text_input("Escribe tu prompt aquí")
 
 if prompt:
-    response = model.generate(prompt)
+    response = model.generate_content(prompt)
     st.write(response.text)