Update app.py

app.py

@@ -134,20 +134,13 @@ def show_violin_plot(df):
     },
 }
 
-import streamlit as st
-
-# Predefined analysis tasks and visualization types
-PREDEFINED_ANALYSIS = {
-    # ... (Predefined analysis tasks remain the same)
-}
-
 def generate_streamlit_app_code(app_title, app_subtitle, side_panel_title, analysis_tasks, requirements):
     # Generate Python code for the Streamlit app
     # Here we make sure the analysis functions are correctly indented at the same level as the main function
     analysis_functions_code = "\n    ".join([PREDEFINED_ANALYSIS[task]['Code'].replace("\n", "\n    ") for task in analysis_tasks])
     # Here we ensure that calls to analysis functions are correctly indented within the main function
     analysis_tasks_code = "\n".join([f"        {PREDEFINED_ANALYSIS[task]['Function']}(df)" for task in analysis_tasks])
-
+    
     code = f"""import streamlit as st
 import pandas as pd
 import seaborn as sns
@@ -164,21 +157,20 @@ def main():
         uploaded_file = st.file_uploader("Upload a CSV file", type=["csv"])
 
         if uploaded_file is not None:
-            @st.cache_data
+            @st.cache
             def load_data(uploaded_file):
                 return pd.read_csv(uploaded_file)
 
             df = load_data(uploaded_file)
 
 {analysis_functions_code}
-
+            
             # Analysis task calls
 {analysis_tasks_code}
     except Exception as e:
         st.error(f"An error occurred: {str(e)}")
 
-if __name__ == "__main__":
-    main()
+main()
 """
     return code
 
@@ -211,7 +203,7 @@ def main():
             # Download the generated files
             with open(file_path, "rb") as file:
                 btn = st.download_button(label="Download your Streamlit app", data=file, file_name=file_path)
-
+                
             if requirements:
                 with open("requirements.txt", "rb") as req_file:
                     btn_req = st.download_button(label="Download requirements.txt", data=req_file, file_name="requirements.txt")
@@ -219,4 +211,4 @@ def main():
             st.warning("Please enter a title for your Streamlit app.")
 
 if __name__ == "__main__":
-    main()
+    main()