Update app.py

app.py

@@ -2,7 +2,11 @@ import streamlit as st
 import pandas as pd
 import matplotlib.pyplot as plt
 import seaborn as sns
-from groqflow.groqmodel import GroqModel
+
+try:
+    from groqflow.groqmodel import GroqModel
+except ImportError as e:
+    st.error(f"Failed to import GroqModel. Please ensure all dependencies are installed correctly. Error: {e}")
 
 # Configure page
 st.set_page_config(page_title="Data Augmentation App", layout="wide")
@@ -26,10 +30,13 @@ st.sidebar.title("Upload Your File")
 st.sidebar.markdown("Supported formats: CSV, Excel")
 
 # Load Groq API key from secrets
-groq_api_key = st.secrets["HUGGINGFACE_KEY"]
-
-# Load Groq model
-groq_model = GroqModel("llama3-8b-8192", api_key=groq_api_key)
+try:
+    groq_api_key = st.secrets["HUGGINGFACE_KEY"]
+    groq_model = GroqModel("llama3-8b-8192", api_key=groq_api_key)
+except KeyError:
+    st.error("API key not found in secrets. Please configure your `HUGGINGFACE_KEY` in Streamlit secrets.")
+except Exception as e:
+    st.error(f"Error initializing GroqModel: {e}")
 
 def load_file(uploaded_file):
     """Load the uploaded file."""
@@ -60,6 +67,9 @@ def generate_graph(data, query):
 def handle_query(data, query):
     """Handle user query using Groq API."""
     try:
+        if not groq_model:
+            st.error("GroqModel is not initialized. Check for errors in setup.")
+            return
         prompt = f"Given the dataset: {data.to_dict(orient='records')}, answer the following: {query}"
         response = groq_model.generate(prompt)
         st.write("Response:", response)