Update app.py

app.py

@@ -11,33 +11,22 @@ import os
 from mitosheet.streamlit.v1 import spreadsheet
 from pygwalker.api.streamlit import init_streamlit_comm, get_streamlit_html
 
-st.set_page_config(
-    page_title="AI TOOL",
-    layout="wide"
-)
+# Global variable to store uploaded file
+uploaded_file = None
 
 def main():
+    global uploaded_file
     st.sidebar.title("App Options")
     option = st.sidebar.selectbox("Choose an option", ["View Instructions", "View Data","Data Profiling","Tableau AI", "CSV Chatbot"])
 
     if option == "View Instructions":
         show_instructions()
     elif option == "Data Profiling":
-        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
-        else:
-            data_profiling(uploaded_file)
+        data_profiling()
     elif option == "CSV Chatbot":
-        openai_api_key = st.text_input("Enter your OpenAI API Key", type="password")
-        if not openai_api_key:
-            st.warning("You should have an OpenAI API key to continue. Get one at [OpenAI API Keys](https://platform.openai.com/api-keys)")
-            st.stop()
-        os.environ['OPENAI_API_KEY'] = openai_api_key
-        personal_assistant()
+        csv_chatbot()
     elif option == "View Data":
-        virtual_excel_sheet()
+        view_data()
     elif option == "Tableau AI":
         tableau_ai()
 
@@ -50,36 +39,52 @@ def show_instructions():
     st.write("- **CSV Chatbot:** Interact with a chatbot to get insights from your CSV data.")
     st.write("- **Tableau AI:** Upload a CSV file to visualize it using Tableau AI.")
     st.write("- **View Instructions:** View these instructions again.")
-    st.markdown(
-        """
-        <a href="https://www.linkedin.com/in/your_linkedin_profile/" target="_blank">
-            <img src="https://github.com/dheereshagrwal/colored-icons/blob/master/public/icons/linkedin/linkedin.svg" width="30" height="30"/>
-        </a>
-        <a href="https://github.com/your_github_username" target="_blank">
-            <img src="https://img.icons8.com/fluency/48/000000/github.png" width="30" height="30"/>
-        </a>
-        """,
-        unsafe_allow_html=True
-    )
-def data_profiling(uploaded_file):
+
+def data_profiling():
+    global uploaded_file
     st.title("Data Profiling App")
+    if uploaded_file is None:
+        uploaded_file = st.file_uploader("Upload CSV file", type=["csv", "xlsx"])
+        if uploaded_file is None:
+            st.warning("Please upload a CSV or Excel file.")
+            st.stop()  # Stop execution if no file uploaded
+
+    if uploaded_file.name.endswith('.xlsx'):
+        # Load Excel file into pandas DataFrame
+        df_excel = pd.read_excel(uploaded_file)
+        # Save DataFrame as CSV
+        csv_filename = uploaded_file.name.replace('.xlsx', '.csv')
+        df_excel.to_csv(csv_filename, index=False)
+        st.success(f"Excel file converted to CSV: {csv_filename}")
+        # Set uploaded file to the converted CSV file
+        uploaded_file = open(csv_filename, 'rb')
+
     df = pd.read_csv(uploaded_file)
     st.dataframe(df)
+
     # Generate and display the data profile report
     pr = ProfileReport(df, title="Report")
     st_profile_report(pr)
 
-def personal_assistant():
+def csv_chatbot():
+    global uploaded_file
     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
+        uploaded_file = st.file_uploader("Upload CSV file", type=["csv", "xlsx"])
+        if uploaded_file is None:
+            st.warning("Please upload a CSV or Excel file.")
+            st.stop()  # Stop execution if no file uploaded
+
+    openai_api_key = st.text_input("Enter your OpenAI API Key", type="password")
+    if not openai_api_key:
+        st.warning("You should have an OpenAI API key to continue. Get one at [OpenAI API Keys](https://platform.openai.com/api-keys)")
+        st.stop()
 
+    os.environ['OPENAI_API_KEY'] = openai_api_key
     llm = OpenAI(temperature=0)
     agent = create_csv_agent(
         llm,
@@ -107,12 +112,24 @@ def personal_assistant():
         st.write(answer)
         st.write(f"Answer (took {round(end - start, 2)} s.)")
 
-def virtual_excel_sheet():
+def view_data():
+    global uploaded_file
     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
+        uploaded_file = st.file_uploader("Upload CSV file", type=["csv", "xlsx"])
+        if uploaded_file is None:
+            st.warning("Please upload a CSV or Excel file.")
+            st.stop()  # Stop execution if no file uploaded
+
+    if uploaded_file.name.endswith('.xlsx'):
+        # Load Excel file into pandas DataFrame
+        df_excel = pd.read_excel(uploaded_file)
+        # Save DataFrame as CSV
+        csv_filename = uploaded_file.name.replace('.xlsx', '.csv')
+        df_excel.to_csv(csv_filename, index=False)
+        st.success(f"Excel file converted to CSV: {csv_filename}")
+        # Set uploaded file to the converted CSV file
+        uploaded_file = open(csv_filename, 'rb')
 
     df = pd.read_csv(uploaded_file)
 
@@ -123,6 +140,7 @@ def virtual_excel_sheet():
     final_dfs, code = spreadsheet(dataframe)
 
 def tableau_ai():
+    global uploaded_file
     st.title("Virtual Tableau AI Tool")
     init_streamlit_comm()
 
@@ -134,11 +152,26 @@ def tableau_ai():
 
     # 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
 
+    if uploaded_file is None:
+        uploaded_file = st.file_uploader("Upload a CSV file", type=["csv", "xlsx"])
+        if uploaded_file is None:
+            st.warning("Please upload a CSV or Excel file.")
+            st.stop()  # Stop execution if no file uploaded
+
+    if uploaded_file.name.endswith('.xlsx'):
+        # Load Excel file into pandas DataFrame
+        df_excel = pd.read_excel(uploaded_file)
+        # Save DataFrame as CSV
+        csv_filename = uploaded_file.name.replace('.xlsx', '.csv')
+        df_excel.to_csv(csv_filename, index=False)
+        st.success(f"Excel file converted to CSV: {csv_filename}")
+        # Set uploaded file to the converted CSV file
+        uploaded_file = open(csv_filename, 'rb')
+
     df = get_user_uploaded_data()
 
     if df is not None: