Update app.py

app.py

@@ -33,13 +33,12 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
 
 # Streamlit interface
 st.title("Chat with Codestral Model")
-system_message = st.text_input("System message", value="You are an expert python coder with in depth knowledge of langchain.")
+system_message = st.text_input("System message", value="You are an expert python coder with in-depth knowledge of langchain.")
 max_tokens = st.slider("Max new tokens", min_value=1, max_value=2048, value=2048, step=1)
 temperature = st.slider("Temperature", min_value=0.0, max_value=1.0, value=0.6, step=0.1)
 top_p = st.slider("Top-p (nucleus sampling)", min_value=0.1, max_value=1.0, value=0.95, step=0.05)
 
-history = []
-
+# Initialize history in session state
 if "history" not in st.session_state:
     st.session_state.history = []
 
@@ -52,7 +51,11 @@ def get_response():
         for r in response_generator:
             response = r
         st.session_state.history[-1] = (user_input, response)
+        st.session_state.user_input = ""  # Clear input after sending
 
-st.text_area("Chat History", value="\n".join([f"User: {h[0]}\nAssistant: {h[1]}" for h in st.session_state.history]), height=300)
+# Display chat history
+st.text_area("Chat History", value="\n".join([f"User: {h[0]}\nAssistant: {h[1]}" for h in st.session_state.history]), height=300, key="chat_history")
 
+# User input
 st.text_input("Your message:", key="user_input", on_change=get_response)
+