Update app.py

app.py

@@ -173,6 +173,37 @@ main()  # Call the main function to run the app
     return code
 
 
+# Define required libraries for each predefined analysis task
+TASK_REQUIREMENTS = {
+    "Basic Statistics": ["pandas", "streamlit"],
+    "Correlation Heatmap": ["pandas", "streamlit", "seaborn"],
+    "Histogram": ["pandas", "streamlit", "plotly"],
+    "Box Plot": ["pandas", "streamlit", "plotly"],
+    "Scatter Plot": ["pandas", "streamlit", "plotly"],
+    "Line Plot": ["pandas", "streamlit", "plotly"],
+    "Bar Chart": ["pandas", "streamlit", "plotly"],
+    "Pair Plot": ["pandas", "streamlit", "seaborn"],
+    "Distribution Plot": ["pandas", "streamlit", "seaborn"],
+    "Count Plot": ["pandas", "streamlit", "seaborn"],
+    "Pie Chart": ["pandas", "streamlit", "plotly"],
+    "Area Plot": ["pandas", "streamlit", "plotly"],
+    "Violin Plot": ["pandas", "streamlit", "plotly"]
+}
+
+def generate_requirements(selected_tasks):
+    # Set to store all unique required libraries
+    required_libraries = set()
+
+    # Add libraries for each selected task
+    for task in selected_tasks:
+        required_libraries.update(TASK_REQUIREMENTS.get(task, []))
+
+    # Convert set to sorted list for consistent order
+    requirements_list = sorted(list(required_libraries))
+
+    # Format list into string to place in requirements.txt format
+    return "\n".join(requirements_list)
+
 def main():
     st.title("Streamlit App Generator")
 
@@ -184,8 +215,9 @@ def main():
     # Select predefined analysis tasks
     selected_tasks = st.multiselect("Select predefined analysis tasks to include:", list(PREDEFINED_ANALYSIS.keys()))
 
-    # Generate requirements.txt file
-    requirements = st.text_area("Enter requirements.txt content (optional):")
+    # Automatically generate requirements.txt content based on selected tasks
+    auto_requirements = generate_requirements(selected_tasks)
+    requirements = st.text_area("Enter requirements.txt content (optional):", value=auto_requirements)
 
     if st.button("Generate and Download"):
         if app_title:
@@ -195,17 +227,15 @@ def main():
                 f.write(generate_streamlit_app_code(app_title, app_subtitle, side_panel_title, selected_tasks, requirements))
 
             # Write requirements.txt file
-            if requirements:
-                with open("requirements.txt", "w") as req_file:
-                    req_file.write(requirements)
+            with open("requirements.txt", "w") as req_file:
+                req_file.write(requirements)
 
             # 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")
+            with open("requirements.txt", "rb") as req_file:
+                btn_req = st.download_button(label="Download requirements.txt", data=req_file, file_name="requirements.txt")
         else:
             st.warning("Please enter a title for your Streamlit app.")