Update app.py

app.py

@@ -88,6 +88,7 @@ class Llama3Demo:
         return response.split("<|assistant|>")[-1].strip()
 
 ##################################################################        
+
 def main():
     st.set_page_config(page_title="Llama 3.2 Chat", page_icon="🦙")
     
@@ -107,44 +108,6 @@ def main():
         except Exception as e:
             st.error(f"Error en configuración: {str(e)}")
     
-    # Inicializar el modelo
-    if 'llama' not in st.session_state:
-        with st.spinner("Inicializando Llama 3.2... esto puede tomar unos minutos..."):
-            try:
-                st.session_state.llama = Llama3Demo()
-            except Exception as e:
-                st.error("Error inicializando el modelo")
-                st.stop()
-    
-    # Gestión del historial de chat
-    if 'messages' not in st.session_state:
-        st.session_state.messages = []
-    
-    # Mostrar historial
-    for message in st.session_state.messages:
-        with st.chat_message(message["role"]):
-            st.markdown(message["content"])
-    
-    # Interface de chat
-    if prompt := st.chat_input("Escribe tu mensaje aquí"):
-        with st.chat_message("assistant"):
-            try:
-                response = st.session_state.llama.generate_response(
-                    prompt,
-                    **generation_params
-                )
-                st.markdown(response)
-            except Exception as e:
-                st.error(f"Error: {str(e)}")
-
-        with st.chat_message("assistant"):
-            try:
-                response = st.session_state.llama.generate_response(prompt)
-                st.markdown(response)
-                st.session_state.messages.append({"role": "assistant", "content": response})
-            except Exception as e:
-                st.error(f"Error generando respuesta: {str(e)}")
-    
     # Sidebar con controles de generación
     with st.sidebar:
         st.markdown("### Parámetros de Generación")
@@ -199,6 +162,43 @@ def main():
         - **Top-p**: Control sobre la variabilidad de respuestas
         - **Longitud**: Ajustar según necesidad de detalle
         """)
+        
+        if st.button("Limpiar Chat"):
+            st.session_state.messages = []
+            st.experimental_rerun()
+    
+    # Inicializar el modelo
+    if 'llama' not in st.session_state:
+        with st.spinner("Inicializando Llama 3.2... esto puede tomar unos minutos..."):
+            try:
+                st.session_state.llama = Llama3Demo()
+            except Exception as e:
+                st.error("Error inicializando el modelo")
+                st.stop()
+    
+    # Gestión del historial de chat
+    if 'messages' not in st.session_state:
+        st.session_state.messages = []
+    
+    # Mostrar historial
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"]):
+            st.markdown(message["content"])
+    
+    # Interface de chat
+    if prompt := st.chat_input("Escribe tu mensaje aquí"):
+        st.session_state.messages.append({"role": "user", "content": prompt})
+        with st.chat_message("user"):
+            st.markdown(prompt)
+
+        with st.chat_message("assistant"):
+            try:
+                response = st.session_state.llama.generate_response(prompt, **generation_params)
+                st.markdown(response)
+                st.session_state.messages.append({"role": "assistant", "content": response})
+            except Exception as e:
+                st.error(f"Error generando respuesta: {str(e)}")
 
 if __name__ == "__main__":
-    main()
+    main()
+