Spaces:
Sleeping
Sleeping
File size: 2,617 Bytes
e58a576 fd6c916 e58a576 fd6c916 e58a576 fd6c916 e58a576 fd6c916 b74c18d fd6c916 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 |
# Import necessary libraries
import os
import streamlit as st
import pandas as pd
from groq import Groq
from PIL import Image
from io import BytesIO
import requests
from streamlit_lottie import st_lottie # Import streamlit-lottie
# Setup environment variables (set your API keys here)
os.environ["GROQ_API_KEY"] = "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"
# Initialize Groq client
client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
# Load dataset once using caching
@st.cache_data
def load_data():
# Load the CSV file only once
data = pd.read_csv('movie_dataset.csv') # Change the path here
return data
# Load Groq model for chat completion
def get_groq_completion(query, model="gemma2-9b-it"):
chat_completion = client.chat.completions.create(
messages=[
{
"role": "user",
"content": query,
}
],
model=model,
)
return chat_completion.choices[0].message.content
# Function to load Lottie animations
def load_lottie_url(url: str):
r = requests.get(url)
if r.status_code != 200:
return None
return r.json()
# Define Streamlit app
def main():
# Page Configuration
st.set_page_config(page_title="Movie RAG Chatbot", page_icon="🎥", layout="wide", initial_sidebar_state="expanded")
# Apply CSS styling for dark theme and animations
st.markdown("""
<style>
body {background-color: #1f1f1f; color: #f5f5f5;}
.stButton>button {background-color: #f5a623; color: white;}
.stTextInput>div>div>input {background-color: #333; color: #f5f5f5;}
h1 {color: #f5a623; text-align: center;}
</style>
""", unsafe_allow_html=True)
st.title("🎬 Movie RAG Chatbot")
# Load the dataset only once
data = load_data()
st.write("### Dataset Overview", data.head())
# Load Lottie animation
lottie_animation_url = "https://assets5.lottiefiles.com/packages/lf20_OZ6W7g.json" # Replace with your desired Lottie URL
lottie_json = load_lottie_url(lottie_animation_url)
if lottie_json:
st_lottie(lottie_json, speed=1, width=None, height=None, key="loading")
# Query input
user_query = st.text_input("Ask about movies or actors:", placeholder="e.g., Tell me about sci-fi movies.")
if st.button("Get Response"):
# Fetch info from RAG model
groq_response = get_groq_completion(user_query)
st.write("### Chatbot Response")
st.markdown(f"Groq Model: {groq_response}")
# Run the main app
if __name__ == "__main__":
main() |