Update app.py

app.py

@@ -2,27 +2,64 @@
 
 import streamlit as st
 import os
-from google.generativeai import configure, GenerativeModel
+import google.generativeai as genai
 from dotenv import load_dotenv
-import fitz  # Para leer PDF si lo usarás más adelante
-import base64
+import fitz  # PyMuPDF para leer PDFs
 
-# Cargar las variables del entorno
+# Cargar variables de entorno
 load_dotenv()
-GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
 
-# Configurar Gemini API
-configure(api_key=GEMINI_API_KEY)
+# Obtener API Key de Gemini
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 
-# Crear el modelo
-model = GenerativeModel('gemini-pro')
+# Configurar la API de Gemini
+genai.configure(api_key=GEMINI_API_KEY)
 
-st.title("Gemini chatbot")
-st.subheader("Hola, hazle una pregunta a Gemini :)")
+# Modelo de Gemini
+MODEL_NAME = "gemini-1.5-pro"
 
-# Input del usuario
-prompt = st.text_input("Escribe tu prompt aquí")
+# Función para generar contenido con Gemini
+def generate_response(user_input):
+    try:
+        model = genai.GenerativeModel(MODEL_NAME)
+        response = model.generate_content(user_input)
+        return response.text if response else "No se recibió respuesta."
+    except Exception as e:
+        return f"Error en la generación de contenido: {str(e)}"
 
-if prompt:
-    response = model.generate_content(prompt)
-    st.write(response.text)
+# Función para extraer texto de un PDF
+def extract_text_from_pdf(pdf_file):
+    try:
+        doc = fitz.open(stream=pdf_file.read(), filetype="pdf")
+        text = "\n".join([page.get_text() for page in doc])
+        return text if text else "No se pudo extraer texto del PDF."
+    except Exception as e:
+        return f"Error al leer el PDF: {str(e)}"
+
+# Interfaz con Streamlit
+st.set_page_config(page_title="Generador con Gemini API", layout="centered")
+st.title("Chat con Gemini API")
+
+# Entrada del usuario (texto manual)
+user_input = st.text_area("Escribe algo:", "")
+
+# Opción para subir un archivo PDF
+uploaded_file = st.file_uploader("📄 O sube un PDF", type=["pdf"])
+
+# Botón para generar respuesta
+if st.button("Generar respuesta"):
+    if uploaded_file:
+        st.write("⏳ Extrayendo texto del PDF...")
+        extracted_text = extract_text_from_pdf(uploaded_file)
+        st.text_area("📄 Texto extraído:", extracted_text, height=200)
+        user_input = extracted_text  # Usar el texto extraído para la consulta
+
+    if user_input:
+        st.write("⏳ Procesando con Gemini...")
+        response = generate_response(user_input)
+        st.subheader("🔹 Respuesta de Gemini:")
+        st.write(response)
+    else:
+        st.warning("⚠️ Ingresa un texto o sube un PDF antes de continuar.")
+
+st.write("💡 Desarrollado por Mariana 🚀")