Update src/app.py

src/app.py

@@ -5,6 +5,17 @@ import base64
 from pathlib import Path
 from backend import load_model, ChatModel
 from config import get_model_options
+import uuid
+
+def set_new_message_uuid():
+    st.session_state.message_uuid = uuid.uuid4()
+    logging.info(f'[NEW MESSAGE UUID] {st.session_state.message_uuid}')
+
+def save_messages():
+    messages = chat_model.conversation.messages
+    message_uuid = st.session_state.message_uuid
+    logging.info(f'[Messages save_at:] {messages} com {message_uuid}')
+
 
 st.set_page_config(page_title="Small LLM - Chat", layout="wide")
 
@@ -86,6 +97,7 @@ with st.sidebar:
                 st.session_state.chat_model = chat_model
                 st.session_state.model_info = selected_model_config
                 st.session_state.model_name = model_name
+                set_new_message_uuid()
                 st.success("✅ Modelo carregado!")
                 if "messages" in st.session_state:
                     del st.session_state.messages
@@ -105,6 +117,7 @@ with st.sidebar:
             
             if st.button("🗑️ Limpar Histórico", use_container_width=True):
                 chat_model.clear_history()
+                set_new_message_uuid()
                 if "messages" in st.session_state:
                     del st.session_state.messages
                 st.rerun()
@@ -112,22 +125,7 @@ with st.sidebar:
 # Área principal - Chat
 if "chat_model" not in st.session_state:
     st.info("👈 Use a sidebar para carregar um modelo primeiro.")
-    st.markdown("""
-    ### Modelos disponíveis:
-    
-    **Google Gemma:**
-    - `google/gemma-3-4b-it` - 4 bilhões de parâmetros
-    - `google/gemma-3-1b-it` - 1 bilhão de parâmetros
-    - `google/gemma-3-270m-it` - 270 milhões de parâmetros
-    
-    **Qwen:**
-    - `Qwen/Qwen3-0.6B` - 600 milhões de parâmetros
-    - `Qwen/Qwen2.5-0.5B-Instruct` - 500 milhões (instruct)
-    - `Qwen/Qwen2.5-0.5B` - 500 milhões
-    
-    **Facebook:**
-    - `facebook/MobileLLM-R1-950M` - 950 milhões de parâmetros
-    """)
+    st.markdown("""  """)
 else:
     chat_model = st.session_state.chat_model
     
@@ -171,6 +169,7 @@ else:
                 
                 chat_model.add_assistant_message(full_response)
                 st.session_state.messages.append({"role": "assistant", "content": full_response})
+                save_messages()
                 
             except Exception as e:
                 error_msg = f"Erro na geração: {str(e)}"