Update app.py

app.py

@@ -10,6 +10,7 @@ GEMINI_API_KEY = "AIzaSyCZwFUTPjCIx5reXELxu8Tby2wwu-s_K9o"
 def generate_response(user_input):
     try:
         client = genai.Client(api_key=GEMINI_API_KEY)
+
         model = "gemini-2.0-flash-thinking-exp-01-21"
         contents = [
             types.Content(
@@ -49,43 +50,30 @@ def extract_text_from_pdf(pdf_file):
     except Exception as e:
         return f"Se produjo un error al leer el PDF: {e}"
 
-# Inicializar historial de conversaciones en la sesión
-if "chat_history" not in st.session_state:
-    st.session_state.chat_history = []
-
+# Aplicación Streamlit
 st.title("Gemini LLM App")
 
-# Menú lateral para elegir la opción (chat o PDF)
-option = st.sidebar.radio("Elige una opción:", ("Chat", "PDF"))
+# 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: Chat con historial
-if option == "Chat":
-    st.header("Chat con Gemini")
+# 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í...")
-    
-    if st.button("Enviar"):
+
+    # 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)
-            # Agregar al historial
-            st.session_state.chat_history.append({
-                "usuario": user_input,
-                "respuesta": response
-            })
             st.success("¡Respuesta generada!")
+            st.write(response)
         else:
             st.error("Por favor, introduce una solicitud para continuar.")
 
-    # Mostrar historial de conversaciones
-    st.subheader("Historial de Conversaciones")
-    for idx, msg in enumerate(st.session_state.chat_history):
-        st.markdown(f"**Usuario {idx+1}:** {msg['usuario']}")
-        st.markdown(f"**Gemini:** {msg['respuesta']}")
-        st.markdown("---")
-
 # Opción 2: Subir y preguntar sobre un PDF
-elif option == "PDF":
-    st.header("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")
 
     if uploaded_file is not None:
@@ -93,19 +81,19 @@ elif option == "PDF":
             pdf_text = extract_text_from_pdf(uploaded_file)
 
         if pdf_text:
-            st.success("Texto extraído correctamente.")
+            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)
-            
+
             # 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í...")
-            
+
+            # Botón para enviar la pregunta
             if st.button("Obtener Respuesta"):
                 if user_question.strip():
                     with st.spinner("Generando respuesta..."):
-                        # Combinar el texto del PDF y la pregunta del usuario
-                        combined_input = pdf_text + "\nPregunta: " + user_question
-                        response = generate_response(combined_input)
+                        response = generate_response(pdf_text, user_question)
                     st.success("¡Respuesta generada!")
                     st.write(response)
                 else:
@@ -113,4 +101,4 @@ elif option == "PDF":
         else:
             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.")
+        st.info("Por favor, sube un archivo PDF para empezar.")