Update app2.py

app2.py

@@ -16,11 +16,13 @@ def initialize_llm(model_choice):
         if not groq_api_key:
             st.error("Groq API key is missing. Please set the GROQ_API_KEY environment variable.")
             return None
+        st.success("Using model: llama-3.3-70b (Groq)")
         return ChatGroq(groq_api_key=groq_api_key, model="groq/llama-3.3-70b-versatile")
     elif model_choice == "GPT-4o":
         if not openai_api_key:
             st.error("OpenAI API key is missing. Please set the OPENAI_API_KEY environment variable.")
             return None
+        st.success("Using model: GPT-4o (OpenAI)")
         return OpenAI(api_token=openai_api_key)
 
 def load_dataset_into_session():
@@ -31,7 +33,6 @@ def load_dataset_into_session():
         dataset_name = st.text_input("Enter Hugging Face Dataset Name:", value="HUPD/hupd")
         if st.button("Load Dataset"):
             try:
-                # Using the original parameters
                 dataset = load_dataset(
                     dataset_name, 
                     name="sample", 
@@ -43,14 +44,17 @@ def load_dataset_into_session():
                 st.success(f"Dataset '{dataset_name}' loaded successfully!")
                 st.dataframe(st.session_state.df.head(10))
             except Exception as e:
-                st.error(f"Error: {e}")
+                st.error(f"Error loading dataset: {e}")
     
     elif input_option == "Upload CSV File":
         uploaded_file = st.file_uploader("Upload CSV File:", type=["csv"])
         if uploaded_file:
-            st.session_state.df = pd.read_csv(uploaded_file)
-            st.success("File uploaded successfully!")
-            st.dataframe(st.session_state.df.head(10))
+            try:
+                st.session_state.df = pd.read_csv(uploaded_file)
+                st.success("File uploaded successfully!")
+                st.dataframe(st.session_state.df.head(10))
+            except Exception as e:
+                st.error(f"Error reading file: {e}")
 
 if "df" not in st.session_state:
     st.session_state.df = None
@@ -58,7 +62,7 @@ if "df" not in st.session_state:
 # Streamlit app
 st.title("Chat With Your Dataset Using PandasAI")
 
-# LLM Selection
+# Section 1: LLM Selection
 st.sidebar.title("Choose Your LLM")
 model_choice = st.sidebar.radio(
     "Select a model:",
@@ -71,11 +75,11 @@ llm = initialize_llm(model_choice)
 if not llm:
     st.stop()
 
-# Dataset Loading
-#st.header("Dataset Selection")
+# Section 2: Dataset Loading
+st.header("Dataset Selection")
 load_dataset_into_session()
 
-# Interact with DataFrame if it's loaded
+# Section 3: Query and Interaction
 if st.session_state.df is not None:
     st.subheader("Ask Questions About Your Dataset")
     chat_df = SmartDataframe(st.session_state.df, config={"llm": llm})
@@ -87,8 +91,8 @@ if st.session_state.df is not None:
             st.write("### Response:")
             st.write(response)
 
-            # Check for keywords that indicate the user wants a plot
-            if any(keyword in user_query.lower() for keyword in ["plot", "graph", "draw", "visualize", "chart", "visualizations", "visualise"]):
+            # Check for plot-related keywords
+            if any(keyword in user_query.lower() for keyword in ["plot", "graph", "draw", "visualize", "chart", "visualise"]):
                 st.write("### Generating Plot...")
                 try:
                     chat_df.chat(user_query)