Update app.py

app.py

@@ -0,0 +1,79 @@
+# 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('/content/movie_dataset.csv')  # Replace with the actual file name
+    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}")