Update app.py

app.py

@@ -1,10 +1,8 @@
 # Run Streamlit app
-import os
 import streamlit as st
+import os
 from dotenv import load_dotenv
 from groq import Groq
-from langchain.memory import ConversationBufferMemory
-from langchain.memory.chat_message_histories import ChatMessageHistory
 
 # Load environment variables
 load_dotenv()
@@ -17,68 +15,54 @@ if not api_key:
 
 client = Groq(api_key=api_key)
 
-# Initialize LangChain memory with updated structure
-chat_history = ChatMessageHistory()
-memory = ConversationBufferMemory(
-    chat_memory=chat_history,
-    return_messages=True
-)
-
-# Function to call the Groq API
-def get_groq_response(messages, model="llama3-8b-8192"):
-    try:
-        chat_completion = client.chat.completions.create(messages=messages, model=model)
-        return chat_completion.choices[0].message.content
-    except Exception as e:
-        st.error(f"Error communicating with Groq API: {e}")
-        return "I'm sorry, there was an error processing your request."
-
 # Streamlit app layout
-st.title("Chat with Groq (No OpenAI)")
-st.write("Interact with an intelligent assistant powered by Groq and LangChain memory!")
+st.title("Chat with Groq")
+st.write("Ask anything, and the Groq model will respond!")
 
 # Sidebar for settings
 st.sidebar.title("Settings")
 model_name = st.sidebar.text_input("Model Name", value="llama3-8b-8192")
 
-# Initialize or load session memory
-if "chat_history" not in st.session_state:
-    st.session_state.chat_history = []
+# Main chat interface
+if "messages" not in st.session_state:
+    st.session_state.messages = [
+        {"role": "system", "content": "You are a helpful assistant."}
+    ]
 
-# Display chat history
-for chat in st.session_state.chat_history:
-    if chat["role"] == "user":
-        st.markdown(f"**You:** {chat['content']}")
-    elif chat["role"] == "assistant":
-        st.markdown(f"**Assistant:** {chat['content']}")
+for message in st.session_state.messages:
+    if message["role"] == "user":
+        st.markdown(f"**User:** {message['content']}")
+    elif message["role"] == "assistant":
+        st.markdown(f"**Assistant:** {message['content']}")
 
 # Input box for user query
-user_input = st.text_input("Your Message", key="user_input")
+user_input = st.text_input("Your Message", key="input_box")
 
-# Send button
+# Handle user input and get response
 if st.button("Send") and user_input.strip():
-    # Add user message to memory and session
-    st.session_state.chat_history.append({"role": "user", "content": user_input})
+    # Add user message to session
+    st.session_state.messages.append({"role": "user", "content": user_input})
 
-    # Construct messages from memory
-    messages = memory.load_memory_variables({}).get("chat_history", [])
-    messages.append({"role": "user", "content": user_input})
+    # Call Groq API
+    try:
+        chat_completion = client.chat.completions.create(
+            messages=st.session_state.messages,
+            model=model_name,
+        )
+        assistant_response = chat_completion.choices[0].message.content
 
-    # Get response from Groq
-    response = get_groq_response(messages, model=model_name)
+        # Add assistant message to session
+        st.session_state.messages.append({"role": "assistant", "content": assistant_response})
 
-    # Add response to memory and session
-    st.session_state.chat_history.append({"role": "assistant", "content": response})
-    memory.save_context(
-        {"input": user_input},  # Pass user input as "input"
-        {"output": response}    # Pass assistant response as "output"
-    )
+    except Exception as e:
+        st.error(f"An error occurred: {e}")
 
-    # Clear the input box
-    st.session_state.user_input = ""
+    # Clear input box
+    st.session_state.input_box = ""
 
-# Clear chat history
-if st.sidebar.button("Clear Chat"):
-    st.session_state.chat_history = []
-    memory.clear()
+# Button to clear chat history
+if st.sidebar.button("Clear Chat History"):
+    st.session_state.messages = [
+        {"role": "system", "content": "You are a helpful assistant."}
+    ]
     st.success("Chat history cleared.")