Update app.py

app.py

@@ -10,7 +10,6 @@ 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(
@@ -50,30 +49,43 @@ def extract_text_from_pdf(pdf_file):
     except Exception as e:
         return f"Se produjo un error al leer el PDF: {e}"
 
-# Aplicación Streamlit
+# Inicializar historial de conversaciones en la sesión
+if "chat_history" not in st.session_state:
+    st.session_state.chat_history = []
+
 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"))
+# Menú lateral para elegir la opción (chat o PDF)
+option = st.sidebar.radio("Elige una opción:", ("Chat", "PDF"))
 
-# Opción 1: Hablar con el chat
-if option == "Hablar con el chat":
-    # Entrada del usuario para el chat
+# Opción 1: Chat con historial
+if option == "Chat":
+    st.header("Chat con Gemini")
     user_input = st.text_area("Introduce tu solicitud:", placeholder="Escribe algo aquí...")
-
-    # Botón para enviar la solicitud
-    if st.button("Obtener Respuesta"):
+    
+    if st.button("Enviar"):
         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 == "Subir y preguntar sobre un PDF":
-    # Subir archivo PDF
+elif option == "PDF":
+    st.header("Preguntar sobre un PDF")
     uploaded_file = st.file_uploader("Sube un archivo PDF", type="pdf")
 
     if uploaded_file is not None:
@@ -81,19 +93,19 @@ elif option == "Subir y preguntar sobre un PDF":
             pdf_text = extract_text_from_pdf(uploaded_file)
 
         if pdf_text:
-            st.success("Texto extraído del PDF correctamente.")
-
+            st.success("Texto extraído 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..."):
-                        response = generate_response(pdf_text, user_question)
+                        # Combinar el texto del PDF y la pregunta del usuario
+                        combined_input = pdf_text + "\nPregunta: " + user_question
+                        response = generate_response(combined_input)
                     st.success("¡Respuesta generada!")
                     st.write(response)
                 else: