app.py

import os
import streamlit as st
from groq import Groq

# Set up the Groq API Key
GROQ_API_KEY = "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"
os.environ["GROQ_API_KEY"] = GROQ_API_KEY

# Initialize the Groq client
client = Groq(api_key=GROQ_API_KEY)

# 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.")

# 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):
    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}],
        model="llama3-8b-8192",  # You can replace with the appropriate model name
    )

    response = chat_completion.choices[0].message.content
    return response

# 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)

    # 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)