Update app.py

app.py

@@ -12,11 +12,14 @@ except Exception as e:
     st.error(f"Error initializing Groq client: {e}")
     st.stop()  # Stop execution if client initialization fails
 
-def get_response(prompt):
+def get_response(prompt, chat_history):
     try:
+        # Include chat history in the request
+        messages = [{"role": "user", "content": msg["user"]} if "user" in msg else {"role": "assistant", "content": msg["bot"]} for msg in chat_history]
+        messages.append({"role": "user", "content": prompt})
         chat_completion = client.chat.completions.create(
-            messages=[{"role": "user", "content": prompt}],
-            model="llama-3.3-70b-versatile" # Or any other available model
+            messages=messages,
+            model="llama-3.3-70b-versatile"  # Or any other available model
         )
         return chat_completion.choices[0].message.content
     except Exception as e:
@@ -26,16 +29,29 @@ def get_response(prompt):
 # Streamlit app layout
 st.title("Interactive Chatbot")
 
+# Initialize session state for chat history
+if "chat_history" not in st.session_state:
+    st.session_state.chat_history = []
+
+# Input box for user message
 user_input = st.text_input("You: ", key="user_input")
 
 if user_input:
-    bot_response = get_response(user_input)
-    st.write("Bot: ", bot_response)
-
-if __name__ == "__main__":
-    st.sidebar.title("Settings")
-    st.sidebar.write("API Key Source: " + ("Environment Variable" if os.environ.get("GROQ_API_KEY") else "Directly in Code (Not Recommended)"))
-    if os.environ.get("GROQ_API_KEY"):
-        st.sidebar.success("Groq Client Initialized Successfully!")
-    else:
-        st.sidebar.warning("Using API Key Directly in Code (Not Recommended). Set GROQ_API_KEY Environment Variable for Security.")
+    # Get bot response
+    bot_response = get_response(user_input, st.session_state.chat_history)
+    # Append to chat history
+    st.session_state.chat_history.append({"user": user_input, "bot": bot_response})
+
+# Display chat history
+st.subheader("Conversation")
+for chat in st.session_state.chat_history:
+    st.write(f"You: {chat['user']}")
+    st.write(f"Bot: {chat['bot']}")
+
+# Sidebar information
+st.sidebar.title("Settings")
+st.sidebar.write("API Key Source: " + ("Environment Variable" if os.environ.get("GROQ_API_KEY") else "Directly in Code (Not Recommended)"))
+if os.environ.get("GROQ_API_KEY"):
+    st.sidebar.success("Groq Client Initialized Successfully!")
+else:
+    st.sidebar.warning("Using API Key Directly in Code (Not Recommended). Set GROQ_API_KEY Environment Variable for Security.")