Update app.py

app.py

@@ -4,12 +4,26 @@ from google.genai import types
 from PyPDF2 import PdfReader
 import os
 
-# Configura tu API Key directamente en el código
+# Leer API Key desde variable de entorno
 GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
 
+# Función para extraer texto de un archivo PDF
+def extract_text_from_pdf(pdf_file):
+    try:
+        pdf_reader = PdfReader(pdf_file)
+        text = ""
+        for page in pdf_reader.pages:
+            text += page.extract_text() + "\n"
+        return text
+    except Exception as e:
+        return f"Se produjo un error al leer el PDF: {e}"
+
 # Función para generar respuesta desde la API de Google Gemini
-def generate_response(user_input):
+def generate_response(context, question):
     try:
+        if not GEMINI_API_KEY:
+            return "❌ No se encontró la API Key. Asegúrate de configurarla como secreto en Hugging Face."
+
         client = genai.Client(api_key=GEMINI_API_KEY)
 
         model = "gemini-2.0-flash-thinking-exp-01-21"
@@ -17,7 +31,7 @@ def generate_response(user_input):
             types.Content(
                 role="user",
                 parts=[
-                    types.Part.from_text(text=user_input),
+                    types.Part.from_text(text=f"Contexto: {context}\n\nPregunta: {question}"),
                 ],
             ),
         ]
@@ -40,66 +54,35 @@ def generate_response(user_input):
     except Exception as e:
         return f"Se produjo un error: {e}"
 
-# Función para extraer texto de un archivo PDF
-def extract_text_from_pdf(pdf_file):
-    try:
-        pdf_reader = PdfReader(pdf_file)
-        text = ""
-        for page in pdf_reader.pages:
-            text += page.extract_text() + "\n"
-        return text
-    except Exception as e:
-        return f"Se produjo un error al leer el PDF: {e}"
-
 # Aplicación Streamlit
-st.title("Gemini LLM App")
-
-# Selección entre Chat o PDF
-option = st.radio("Elige una opción:", ("Hablar con el chat", "Subir y preguntar sobre un PDF"))
-
-# Opción 1: Hablar con el chat
-if option == "Hablar con el chat":
-    # Entrada del usuario para el chat
-    user_input = st.text_area("Introduce tu solicitud:", placeholder="Escribe algo aquí...")
-
-    # Botón para enviar la solicitud
-    if st.button("Obtener Respuesta"):
-        if user_input.strip():
-            with st.spinner("Generando respuesta..."):
-                response = generate_response(user_input)
-            st.success("¡Respuesta generada!")
-            st.write(response)
-        else:
-            st.error("Por favor, introduce una solicitud para continuar.")
+st.title("Gemini LLM App - Consulta PDFs")
 
-# Opción 2: Subir y preguntar sobre un PDF
-elif option == "Subir y preguntar sobre un PDF":
-    # Subir archivo PDF
-    uploaded_file = st.file_uploader("Sube un archivo PDF", type="pdf")
+# Subir archivo PDF
+uploaded_file = st.file_uploader("Sube un archivo PDF", type="pdf")
 
-    if uploaded_file is not None:
-        with st.spinner("Extrayendo texto del PDF..."):
-            pdf_text = extract_text_from_pdf(uploaded_file)
+if uploaded_file is not None:
+    with st.spinner("Extrayendo texto del PDF..."):
+        pdf_text = extract_text_from_pdf(uploaded_file)
 
-        if pdf_text:
-            st.success("Texto extraído del PDF correctamente.")
+    if pdf_text:
+        st.success("Texto extraído del PDF correctamente.")
 
-            # Mostrar una parte del contenido extraído
-            st.text_area("Contenido del PDF:", pdf_text[:1000], height=200)
+        # Mostrar una parte del contenido extraído
+        st.text_area("Contenido del PDF:", pdf_text[:1000], height=200)
 
-            # Entrada de pregunta del usuario sobre el PDF
-            user_question = st.text_area("Haz una pregunta sobre el contenido del PDF:", placeholder="Escribe tu pregunta aquí...")
+        # Entrada de pregunta del usuario
+        user_question = st.text_area("Haz una pregunta sobre el contenido del PDF:", placeholder="Escribe tu pregunta aquí...")
 
-            # Botón para enviar la pregunta
-            if st.button("Obtener Respuesta"):
-                if user_question.strip():
-                    with st.spinner("Generando respuesta..."):
-                        response = generate_response(pdf_text, user_question)
-                    st.success("¡Respuesta generada!")
-                    st.write(response)
-                else:
-                    st.error("Por favor, introduce una pregunta para continuar.")
-        else:
-            st.error("No se pudo extraer texto del archivo PDF. Por favor, intenta con otro archivo.")
+        # Botón para enviar la pregunta
+        if st.button("Obtener Respuesta"):
+            if user_question.strip():
+                with st.spinner("Generando respuesta..."):
+                    response = generate_response(pdf_text, user_question)
+                st.success("¡Respuesta generada!")
+                st.write(response)
+            else:
+                st.error("Por favor, introduce una pregunta para continuar.")
     else:
-        st.info("Por favor, sube un archivo PDF para empezar.")
+        st.error("No se pudo extraer texto del archivo PDF. Por favor, intenta con otro archivo.")
+else:
+    st.info("Por favor, sube un archivo PDF para empezar.")