Update app.py

app.py

@@ -1,11 +1,21 @@
 import streamlit as st
 import os
 
-def generate_streamlit_app_code(app_title, app_content):
+# Predefined code snippets for common components
+PREDEFINED_COMPONENTS = {
+    "Title": "st.title('Your Title Here')",
+    "Subtitle": "st.subheader('Your Subtitle Here')",
+    "Side Panel": "st.sidebar.title('Sidebar Title')",
+    # Add more predefined components as needed
+}
+
+def generate_streamlit_app_code(app_title, app_content, requirements):
     # Generate Python code for the Streamlit app
     code = f"""
 import streamlit as st
 
+{requirements}
+
 def main():
     st.title("{app_title}")
     {app_content}
@@ -21,20 +31,37 @@ def main():
     app_title = st.text_input("Enter your Streamlit app title:")
     app_content = st.text_area("Enter your Streamlit app content (Python code):")
 
+    # Include predefined components
+    selected_components = st.multiselect("Select predefined components to include:", list(PREDEFINED_COMPONENTS.keys()))
+
+    # Generate requirements.txt file
+    requirements = st.text_area("Enter requirements.txt content (optional):")
+
     if st.button("Generate and Download"):
         if app_title and app_content:
-            app_code = generate_streamlit_app_code(app_title, app_content)
+            # Include selected predefined components in the app content
+            for component in selected_components:
+                app_content += f"\n{PREDEFINED_COMPONENTS[component]}"
 
             # Write generated code to a .py file
             file_path = f"{app_title.replace(' ', '_').lower()}_app.py"
             with open(file_path, "w") as f:
-                f.write(app_code)
+                f.write(generate_streamlit_app_code(app_title, app_content, requirements))
+
+            # Write requirements.txt file
+            if requirements:
+                with open("requirements.txt", "w") as req_file:
+                    req_file.write(requirements)
 
-            # Download the generated file
+            # 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")
         else:
             st.warning("Please enter a title and content for your Streamlit app.")
 
 if __name__ == "__main__":
-    main()
+    main()