Update app.py

app.py

@@ -1,8 +1,8 @@
-import os
 import streamlit as st
 from huggingface_hub import InferenceClient
 
-st.title("MAYA 3.0 CHATBOT IA")
+st.set_page_config(page_title="MAYA 3.0 CHATBOT IA", page_icon=":robot_face:")
+
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 conversation_history = []
 
@@ -33,22 +33,44 @@ def format_prompt(message, history):
     prompt += f"[INST] {message} [/INST]"
     return prompt
 
-with st.container():
-    user_input = st.text_input("Usuario:")
-    system_prompt = st.text_input("System Prompt:")
-    temperature = st.slider("Temperatura", 0.0, 1.0, 0.9)
-    max_new_tokens = st.slider("Número máximo de nuevos tokens", 0, 2048, 2048)
-    top_p = st.slider("Top-p (nucleus sampling)", 0.0, 1.0, 0.95)
-    repetition_penalty = st.slider("Penalización por repetición", 1.0, 2.0, 1.0)
+def validate_input(user_input, system_prompt):
+    if not user_input:
+        st.warning("Por favor ingrese un mensaje.")
+        return False
+    if not system_prompt:
+        st.warning("Por favor ingrese un prompt del sistema.")
+        return False
+    return True
+
+def display_help():
+    st.write("También puede ingresar un prompt del sistema para guiar la generación de texto.")
+
+def display_conversation_history(history):
+    for role, message in history:
+        if role == "user":
+            st.text(f"Usuario: {message}")
+        elif role == "bot":
+            st.text(f"Chatbot: {message}")
+
+def main():
+    st.title("MAYA 3.0 CHATBOT IA")
+    user_input = st.text_input("Escribe tu mensaje")
+    system_prompt = st.text_input("Ingrese el prompt del sistema")
+    temperature = st.hidden_selectbox("Temperatura", options=list(range(1, 11)))
+    max_new_tokens = st.hidden_selectbox("Número máximo de nuevos tokens", options=list(range(1, 2049)))
+    top_p = st.hidden_selectbox("Top-p (nucleus sampling)", options=list(range(1, 11)))
+    repetition_penalty = st.hidden_selectbox("Penalización por repetición", options=list(range(1, 2)))
 
     if st.button("Enviar"):
-        if user_input:
+        if validate_input(user_input, system_prompt):
             conversation_history.append(("user", user_input))
             bot_response = generate_response(user_input, conversation_history, system_prompt, temperature, max_new_tokens, top_p, repetition_penalty)
-            conversation_history.append(("bot", bot_response))
+            conversation_history.append(("bot", bot_response)
+
+    if st.button("Ayuda"):
+        display_help()
+
+    display_conversation_history(conversation_history)
 
-for role, message in conversation_history:
-    if role == "user":
-        st.text(f"Usuario: {message}")
-    elif role == "bot":
-        st.text(f"Chatbot: {message}")
+if __name__ == "__main__":
+    main()