Update app.py

app.py

@@ -1,9 +1,36 @@
+%%writefile app.py
+import pandas as pd
+from ydata_profiling import ProfileReport
 import streamlit as st
-import os
+from streamlit_pandas_profiling import st_profile_report
+from langchain.llms.openai import OpenAI
+from langchain.agents.agent_types import AgentType
+import time
 
 def main():
-    st.sidebar.title("OpenAI Settings")
+    st.sidebar.title("App Options")
+    option = st.sidebar.selectbox("Choose an option", ["Data Profiling", "Personal Assistant"])
+
+    if option == "Data Profiling":
+        data_profiling()
+    elif option == "Personal Assistant":
+        personal_assistant()
+
+def data_profiling():
+    st.title("Data Profiling App")
+
+    # Load the data
+    df = pd.read_csv('data.csv')
+
+    # Display the dataframe
+    st.dataframe(df)
 
+    # Generate and display the data profile report
+    pr = ProfileReport(df, title="Report")
+    st_profile_report(pr)
+
+def personal_assistant():
+    st.sidebar.title("OpenAI Settings")
     openai_api_key = st.sidebar.text_input("Enter your OpenAI API Key", type="password")
 
     os.environ['OPENAI_API_KEY'] = openai_api_key
@@ -12,18 +39,10 @@ def main():
     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
-    
-    
-    from langchain.llms.openai import OpenAI
-    from langchain.agents.agent_types import AgentType
-    #from langchain.agents import create_csv_agent
-    from langchain_experimental.agents import create_csv_agent
-    import time
 
     llm = OpenAI(temperature=0)
     agent = create_csv_agent(
@@ -44,8 +63,5 @@ def main():
             st.write(answer)
             st.write(f"Answer (took {round(end - start, 2)} s.)")
 
-    
-
 if __name__ == "__main__":
     main()
-