Update app.py

app.py

@@ -2,19 +2,37 @@ import os
 import streamlit as slt
 from langchain_groq import ChatGroq
 api_key = os.getenv("pack1st")
-def load_answer(question):
-  llms = ChatGroq(model_name = 'gemma2-9b-it', api_key=api_key)
-  answer = llms.invoke(question)
-  return answer.content
-slt.set_page_config(page_title='Basic Langchain model', page_icon=':robot:')
-slt.header('Langchain Model')
-def get_text():
-  input_text = slt.text_input('You: ', key='input')
-  return input_text
-user_input=get_text()
-response = load_answer(user_input)
-submit = slt.button('Generate')
-
-if submit:
-  slt.subheader('Chatbot:')
-  slt.write(response)
+
+
+if not api_key:
+    st.error("GROQ_API_KEY is not set. Please add it to your Hugging Face Space Secrets.")
+    st.stop()
+
+
+llm = ChatGroq(model_name="gemma2-9b-it", api_key=api_key)
+
+
+st.set_page_config(page_title="Basic Langchain Model", page_icon=":robot:")
+st.header("Langchain Model Chatbot")
+
+
+if "messages" not in st.session_state:
+    st.session_state.messages = []  
+
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]): 
+        st.write(message["content"])
+
+if prompt := st.chat_input("Type your message..."):
+    
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    with st.chat_message("user"):
+        st.write(prompt)
+
+    
+    with st.chat_message("assistant"):
+        response = llm.invoke(prompt).content 
+        st.write(response)
+
+    
+    st.session_state.messages.append({"role": "assistant", "content": response})