Update app.py

app.py

@@ -9,18 +9,18 @@ os.environ["GROQ_API_KEY"] = GROQ_API_KEY
 # Initialize the Groq client
 client = Groq(api_key=GROQ_API_KEY)
 
-# Streamlit user interface
-st.title("AI Study Assistant Chatbot")
-st.write("Hello! I'm your AI Study Assistant. You can ask me anything related to your studies or exam preparation.")
+# Streamlit user interface setup
+st.title("AI Chatbot")
+st.write("Hello! I'm your AI Study Assistant. Ask me anything, and I'll try to help.")
 
-# User input for conversation
-user_input = st.text_input("Ask me anything about your study plan or exam:")
+# Initialize session state for maintaining conversation
+if 'conversation_history' not in st.session_state:
+    st.session_state.conversation_history = []
 
 # Function to generate chatbot response based on user input
 def generate_chatbot_response(user_message):
-    # Generate prompt for Groq based on user message
     prompt = f"You are a helpful AI chatbot. The user is asking: {user_message}. Provide a detailed, helpful response."
-
+    
     # Generate response using Groq API
     chat_completion = client.chat.completions.create(
         messages=[{"role": "user", "content": prompt}],
@@ -30,10 +30,17 @@ def generate_chatbot_response(user_message):
     response = chat_completion.choices[0].message.content
     return response
 
-# Display chatbot response when user submits a message
+# User input for conversation
+user_input = st.text_input("Ask me anything:")
+
+# Handle user input and display conversation
 if user_input:
     chatbot_response = generate_chatbot_response(user_input)
-    st.write("### Chatbot Response:")
-    st.write(chatbot_response)
-else:
-    st.write("Please ask me a question about your study plan or exam.")
+
+    # Save the conversation history
+    st.session_state.conversation_history.append(("User: " + user_input, "Chatbot: " + chatbot_response))
+
+# Display chat history
+for question, answer in st.session_state.conversation_history:
+    st.markdown(f"**{question}**")
+    st.markdown(answer)