Update app.py

app.py

@@ -43,3 +43,30 @@ llm = ChatGroq(
     max_tokens=200,
     callbacks=[LLMCallbackHandler(Path("prompts.jsonl"))],
 )
+
+st.title("Blah Blah App Using CrewAI 🚀")
+st.write("Analyze datasets using natural language queries powered by SQL and CrewAI.")
+
+# Initialize session state for data persistence
+if "df" not in st.session_state:
+    st.session_state.df = None
+
+# Dataset Input
+input_option = st.radio("Select Dataset Input:", ["Use Hugging Face Dataset", "Upload CSV File"])
+if input_option == "Use Hugging Face Dataset":
+    dataset_name = st.text_input("Enter Hugging Face Dataset Name:", value="Einstellung/demo-salaries")
+    if st.button("Load Dataset"):
+        try:
+            with st.spinner("Loading dataset..."):
+                dataset = load_dataset(dataset_name, split="train")
+                st.session_state.df = pd.DataFrame(dataset)
+                st.success(f"Dataset '{dataset_name}' loaded successfully!")
+                st.dataframe(st.session_state.df.head())
+        except Exception as e:
+            st.error(f"Error: {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())