Update app.py

app.py

@@ -53,18 +53,88 @@ def show_instructions():
 
 def data_profiling(uploaded_file):
     st.title("Data Profiling App")
-
-    # Load the data
     df = pd.read_csv(uploaded_file)
-
-    # Display the dataframe
     st.dataframe(df)
-
     # Generate and display the data profile report
     pr = ProfileReport(df, title="Report")
     st_profile_report(pr)
 
-# Define other functions (personal_assistant, virtual_excel_sheet, tableau_ai) as before...
+def personal_assistant():
+    st.sidebar.title("OpenAI Settings")
+    st.title("Personal Assistant")
+    st.text("A BR CREATION")
+    st.image("chatbot.jpg", caption="Chatbot", width=178)
+   
+    uploaded_file = st.file_uploader("Upload CSV file", type=["csv"])
+    if uploaded_file is None:
+        st.warning("Please upload a CSV file.")
+        st.stop()  # Stop execution if no file uploaded
+
+    llm = OpenAI(temperature=0)
+    agent = create_csv_agent(
+        llm,
+        uploaded_file,
+        verbose=False,
+        agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
+    )
+
+    predefined_questions = ["How many rows are there in the dataset?", "Explain the dataset."]
+    selected_question = st.selectbox("Select a question", ["Select a question"] + predefined_questions) 
+    custom_question = st.text_input("Or ask a custom question")
+
+    if st.button("Ask"):
+        if selected_question != "Select a question":
+            query = selected_question
+        elif custom_question.strip() != "":
+            query = custom_question.strip()
+        else:
+            st.warning("Please select a predefined question or ask a custom question.")
+            return
+
+        start = time.time()
+        answer = agent.run(query)
+        end = time.time()
+        st.write(answer)
+        st.write(f"Answer (took {round(end - start, 2)} s.)")
+
+def virtual_excel_sheet():
+    st.title("Data Viewer Portal")
+    uploaded_file = st.file_uploader("Upload CSV file", type=["csv"])
+    if uploaded_file is None:
+        st.warning("Please upload a CSV file.")
+        st.stop()  # Stop execution if no file uploaded
+
+    df = pd.read_csv(uploaded_file)
+
+    # Convert the dataframe to a list of dictionaries
+    dataframe = df.to_dict(orient="records")
+
+    # Display the dataframe in a Mito spreadsheet
+    final_dfs, code = spreadsheet(dataframe)
+
+def tableau_ai():
+    st.title("Virtual Tableau AI Tool")
+    init_streamlit_comm()
+
+    # Function to get PygWalker HTML
+    @st.cache_data
+    def get_pyg_html(df: pd.DataFrame) -> str:
+        html = get_streamlit_html(df, use_kernel_calc=True, debug=False)
+        return html
+
+    # Function to get user uploaded DataFrame
+    def get_user_uploaded_data():
+        uploaded_file = st.file_uploader("Upload a CSV file", type=["csv"])
+        if uploaded_file is not None:
+            return pd.read_csv(uploaded_file)
+        return None
+
+    df = get_user_uploaded_data()
+
+    if df is not None:
+        components.html(get_pyg_html(df), width=1300, height=1000, scrolling=True)
+    else:
+        st.write("Please upload a CSV file to proceed.")
 
 if __name__ == "__main__":
     main()