Update app.py

app.py

@@ -1,15 +1,14 @@
 import streamlit as st
 import google.generativeai as genai
 import re
-import datetime
 import os
-import yaml
+import platform
 
 # Secret key and Google Gemini API configuration
 API_KEY = st.secrets["GOOGLE_API_KEY"]
 
 # Page configuration
-st.set_page_config(page_title="⚙️ Gemini2 Puppet Bolt Gen Pro", page_icon="⚙️", layout="wide")
+st.set_page_config(page_title="🐍✨ Gemini2 Pip Gen Pro", page_icon="🐍✨", layout="wide")
 
 # --- Helper Functions ---
 def send_message_to_model(message, model_name, temperature, top_p, top_k, max_tokens):
@@ -34,92 +33,82 @@ def send_message_to_model(message, model_name, temperature, top_p, top_k, max_to
         st.error(f"❌ Error communicating with the AI: {e}")
         return None
 
-def generate_puppet_bolt_plan(prompt_base, detail_level, os_type, security_level, model_name, temperature, top_p, top_k, max_tokens, prompt_detail, encoding, add_header, log_level, bolt_user, bolt_ssh_pass, custom_requirements, specific_details, target_nodes_description, task_type, task_code, parameters_description):
-    """Generates a Puppet Bolt plan based on user settings."""
+def generate_pip_command(prompt_base, model_name, temperature, top_p, top_k, max_tokens, selected_libraries, selected_groups, custom_requirements, specific_details, pip_options, target_os):
+    """Generates a pip command based on user settings."""
     prompt = f"""
-    You are an expert automation engineer specializing in Puppet Bolt. Your task is to generate a complete, secure, and efficient Puppet Bolt plan based on the following description:
-
-    **Goal:** Create the most complete, detailed, efficient, and secure Puppet Bolt plan possible, considering all variables, edge cases, and potential scenarios.
-
-    **Plan Description:** {prompt_base}
-
-    **Detail Level:** {detail_level}
-    **Security Level:** {security_level}
-    **Prompt Detail Level**:{prompt_detail}
-    **Operating System:** {os_type}
-    **Target Nodes:** {target_nodes_description if target_nodes_description else "None"}
-    **Bolt User:** {bolt_user if bolt_user else "root"}
-    **Bolt SSH Password:** {bolt_ssh_pass if bolt_ssh_pass else "None"}
-     **Parameters:** {parameters_description if parameters_description else "None"}
-    **Custom Requirements:** {custom_requirements if custom_requirements else "None"}
-    **Specific Details:** {specific_details if specific_details else "None"}
-    **Task Type:** {task_type}
-    **Task Code:** {task_code if task_code else "None"}
-
-    **Response Format:**
-    - Respond in Markdown format, including a YAML code block with its original formatting, without line breaks.
-    - The YAML code block must be delimited by ```yaml and ```.
-    - Do not include comments, explanations, or any other text outside the code block, unless the prompt requires.
-    - The code must maintain its full vertical formatting, respecting indentation and line breaks.
-    - The code must be realistic, using real-world examples, data, and situations.
-     - The plan must use advanced Bolt resources, such as tasks, parameters, variables, functions, conditionals, loops and targets when necessary.
-      -The plan must use the `run_task` function to execute the tasks.
-      -The plan must use a valid node target.
-    - Explore different approaches, techniques, and advanced practices, always prioritizing security and efficiency.
-     - The plan must use native Bolt functions and modules instead of others.
-    - Unless the user specifies otherwise, use the most current and secure versions of the modules, using bolt and following best practices.
-    - If the description asks to create a file, the plan should create the file directly in the file system and not use a screen output for this.
-    - If the description asks to read a file, the plan should read the file directly from the file system and not use a screen input for this.
-    - Use incremental reasoning to add improvements, expansions, and considerations to your code.
-    - Use the history of the conversations so that the response is incremental.
-
-    **Log Level**:{log_level}
-
-    **Important:**
-    - Generate only one plan at a time.
-    - Create the longest, most complete, and detailed code possible to cover a wide range of possibilities and scenarios.
-    - Consider all the details of the request, expanding the response and improving the configuration.
-     - If the prompt does not include parameters, but if the user add a variable in a code block, create a task parameter to use that variable.
-     - If the prompt includes parameters, create a task parameter to use that parameters.
-    """
+        You are an expert Python development assistant. Your task is to generate a complete and efficient pip command based on the following description:
+
+        **Goal:** Create the most complete, detailed, and efficient pip install command possible, considering all variables, edge cases, and potential scenarios.
+        
+        **Target Operating System:** {target_os}
+
+        **Command Description:** {prompt_base}
+
+        **Selected Libraries:** {selected_libraries if selected_libraries else "None"}
+        **Selected Groups:** {selected_groups if selected_groups else "None"}
+        **Custom Requirements:** {custom_requirements if custom_requirements else "None"}
+        **Specific Details:** {specific_details if specific_details else "None"}
+         **Pip Options:** {pip_options if pip_options else "None"}
+
+        **Response Format:**
+        - Respond in Markdown format, including a pip command code block, a bash code block and powershell code block with its original formatting, without line breaks.
+        - The pip command code block must be delimited by ```pip and ```.
+         - The bash code block must be delimited by ```bash and ```.
+        - The powershell code block must be delimited by ```powershell and ```.
+        - Do not include comments, explanations, or any other text outside the code block.
+        - The code must maintain its full vertical formatting, respecting indentation and line breaks.
+        - Explore different approaches, techniques, and advanced practices, always prioritizing security and efficiency.
+        - Use advanced pip resources such as version constraints, index options, and requirements files when necessary.
+        - If a specific version of a library is specified, use it.
+        - Unless the user specifies otherwise, use the most current versions of the libraries and pip, using and following best practices.
+        - Use incremental reasoning to add improvements, expansions, and considerations to your code.
+        - Use the history of the conversations so that the response is incremental.
+         - If the target OS is Windows, use pip commands compatible with Windows, and if the target OS is Linux, use pip commands compatible with Linux
+
+        **Important:**
+        - Generate only one command at a time.
+        - Create the longest, most complete, and detailed command possible to cover a wide range of possibilities and scenarios.
+        - Consider all the details of the request, expanding the response and improving the command.
+        - If the prompt asks to install a library that is not in a given group, install it in the command, but include in the prompt that library is not in the selected groups.
+        - If a version of a library is specified, install it using a version constraint.
+        - Consider if the user needs a requirements.txt file or not and if needed use it in your command.
+       """
+
     response = send_message_to_model(prompt, model_name, temperature, top_p, top_k, max_tokens)
     return response
 
-def parse_and_save_yml(ai_code, short_title, encoding, add_header):
-    """Parses the markdown and saves the Puppet Bolt plan as .yml."""
-    match = re.search(r'```yaml\s*(.*?)\s*```', ai_code, re.DOTALL | re.IGNORECASE)
-    if match:
-        yml_code = match.group(1).strip()
-    else:
-        yml_code = ai_code.strip()
-        
-    try:
-        yaml.safe_load(yml_code)
-    except yaml.YAMLError as e:
-        st.error(f"❌ Generated YAML is invalid: {e}")
-        return None, None
+def parse_and_save_code(ai_code, short_title):
+    """Parses the markdown and saves the code."""
+    pip_match = re.search(r'```pip\s*(.*?)\s*```', ai_code, re.DOTALL | re.IGNORECASE)
+    bash_match = re.search(r'```bash\s*(.*?)\s*```', ai_code, re.DOTALL | re.IGNORECASE)
+    ps1_match = re.search(r'```powershell\s*(.*?)\s*```', ai_code, re.DOTALL | re.IGNORECASE)
+    
+    pip_code = pip_match.group(1).strip() if pip_match else ai_code.strip()
+    bash_code = bash_match.group(1).strip() if bash_match else None
+    ps1_code = ps1_match.group(1).strip() if ps1_match else None
+
+    base_file_name = f"pip_command_{short_title}"
+
+    files = {}
+    
+    files["pip"]= {"name":f"{base_file_name}.txt", "code":pip_code}
+
+    if bash_code:
+         files["bash"]= {"name":f"{base_file_name}.sh", "code":bash_code}
     
+    if ps1_code:
+       files["powershell"]= {"name":f"{base_file_name}.ps1", "code":ps1_code}
     
-    file_name = f"bolt_{short_title}.yml"
-
-    if add_header:
-        current_date = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-        header = f"""#===============================================================================
-# Puppet Bolt Plan Generated by Google Gemini 2 Bolt Gen Pro
-# Date: {current_date}
-# Author: Elias Andrade AKA Chaos4455
-#===============================================================================\n"""
-        yml_code = header + yml_code
-
-    with open(file_name, "w", encoding=encoding) as f:
-        f.write(yml_code)
-        
-    return file_name, yml_code
+    for key, value in files.items():
+        with open(value["name"], "w") as f:
+            f.write(value["code"])
+
+    return files
 
 
 def main():
-    st.title("⚙️ Gemini2 Puppet Bolt Gen Pro by [Elias Andrade](https://github.com/chaos4455)")
-    st.markdown("Generate advanced Puppet Bolt plans with ease! 🚀")
+    st.title("🐍✨ Gemini2 Pip Gen Pro by [Elias Andrade](https://github.com/chaos4455)")
+    st.markdown("Generate advanced pip install commands with ease! 🚀")
     st.markdown("---")
 
     # Layout in columns (sidebar and main area)
@@ -135,96 +124,94 @@ def main():
             top_k = st.slider("Top K", min_value=1, max_value=100, value=40, step=1, help="Adjust the AI's number of candidate tokens.")
             max_tokens = st.number_input("📏 Max Tokens", min_value=128, max_value=8192, value=8192, step=128, help="Adjust the maximum size of the response.")
             
-        with st.expander("📝 Prompt Settings"):
-             prompt_presets = st.selectbox("🎯 Predefined Prompts", ["None", "Install packages", "Run commands", "Manage services", "Manage files", "System Updates", "Copy Files"], index=0, help="Choose a predefined prompt.")
-             prompt_detail = st.selectbox("🧐 Prompt Detail", ["More descriptive", "Default", "Concise"], index=1, help="Defines the level of detail of the prompt")
-
-        with st.expander("⚙️ Puppet Bolt Settings"):
-            os_type = st.selectbox(
-                "💻 Operating System",
-                [
-                    "Linux", "Windows", "Other"
-                ],
-                index=0, help="Choose the target operating system."
-            )
-            
-            bolt_user = st.text_input("👤 Bolt User", value="root", help="User for Bolt SSH connection")
-            bolt_ssh_pass = st.text_input("🔑 Bolt SSH Password", type="password", help="Password for Bolt SSH connection (Use with caution)")
-
-            encoding = st.selectbox("🔤 Encoding", ["utf-8", "ansi"], index=0, help="Choose the encoding of the .yml file.")
-            add_header = st.checkbox("📜 Add Header", value=True, help="Add a header with information in the .yml file.")
-            log_level = st.selectbox("🗂️ Logging Level", ["Detailed", "Default", "Minimum"], index=1, help="Defines the detail level of logs.")
-            security_level = st.radio("Security Level", ["High", "Medium", "Low"], index=1)
-            custom_requirements = st.text_input("Custom Requirements", placeholder="Ex: Use specific modules, parameters", help="Add specific requirements for the generated plan.")
-            specific_details = st.text_input("Specific Details", placeholder="Ex: Specific configurations, edge cases", help="Add specific details for plan generation.")
-            
-            detail_level = st.selectbox("Detail Level", ["More detailed", "Default", "More concise"], index=1)
-           
-        with st.expander("🔨 Task Settings"):
-            task_type = st.selectbox("🎯 Task Type", ["command", "script", "puppet"], index = 0, help="Type of task")
-            task_code = st.text_area("📋 Task Code",  placeholder="Command or Code for the task", help="Enter your command, script or Puppet code")
-            parameters_description = st.text_input("🧰 Parameters Description",  placeholder="List of parameters for the task", help="Enter the parameters needed for this task")
+        with st.expander("📝 Libraries & Groups"):
+            available_libraries = {
+                 "📊 Data Analysis": ["pandas", "numpy", "scipy", "matplotlib", "seaborn", "plotly", "bokeh", "statsmodels", "scikit-image", "geopandas", "altair", "holoviews", "datashader", "missingno", "vaex", "dask", "xarray", "polars", "arrow", "numba", "cudf", "cupy", "streamz", "panel", "hvplot"],
+                "🧠 Machine Learning": ["scikit-learn", "tensorflow", "torch", "keras", "xgboost", "lightgbm", "catboost", "pytorch-lightning", "transformers", "optuna", "mlflow", "gradio", "huggingface-hub", "sentence-transformers", "fastai", "librosa", "gensim", "spacy", "nltk", "opencv-python", "imbalanced-learn", "sktime", "umap-learn", "fairlearn", "shap", "eli5", "snorkel", "thinc"],
+                "🕸️ Web Development": ["flask", "fastapi", "django", "requests", "beautifulsoup4", "aiohttp", "uvicorn", "gunicorn", "jinja2", "starlette", "websockets", "flask-restful", "django-rest-framework", "scrapy", "selenium", "playwright", "httpx", "rich", "pyramid", "bottle", "tornado", "dash", "plotly-dash", "streamlit", "gradio"],
+                 "🗄️ Database": ["sqlalchemy", "psycopg2", "pymongo", "mysql-connector-python", "sqlite3", "redis", "cassandra-driver", "pyodbc", "aiosqlite", "kafka-python", "motor", "neo4j", "influxdb", "elasticsearch", "pyarrow", "clickhouse-driver", "arangodb", "couchbase", "dgraph-python", "tinydb", "dataset"],
+                "☁️ Cloud": ["boto3", "google-cloud-storage", "azure-storage-blob", "kubernetes", "docker", "apache-libcloud", "pulumi", "awscli", "google-cloud-sdk", "azure-cli", "terraform", "openstack", "ansible", "salt", "chef", "cdktf", "serverless", "aws-cdk", "google-cloud-build", "azure-pipelines", "docker-compose", "moto", "localstack"],
+                 "🤖 IA & LLM": ["langchain", "openai", "diffusers", "sentence-transformers", "stable-diffusion", "huggingface-hub", "pyannote-audio", "whisper", "nltk", "spacy", "transformers", "accelerate", "datasets", "einops", "faiss-cpu", "bitsandbytes", "peft", "trl", "haystack", "llama-index", "deepsparse", "nanogpt", "autogpt", "babyagi","ml-agents", "tensorboardx"],
+                 "🛠️ Dev Tools": ["pytest", "flake8", "mypy", "black", "isort", "pylint", "tox", "pre-commit", "bandit", "coverage", "virtualenv", "pipenv", "poetry", "invoke", "nox", "twine", "wheel", "setuptools", "build", "debugpy", "pyinstrument", "memory-profiler", "cProfile", "pdbpp", "ipython"],
+                "⚙️ Data Engineering": ["apache-airflow", "dask", "luigi", "pyspark", "ray", "prefect", "dbt-core", "pandas-gbq", "petl", "sqlalchemy-redshift", "fugue", "ibis-framework", "mindsdb", "koalas", "vaex-core", "feast", "flink", "beam", "kafka-python", "clickhouse-driver", "superset", "metabase", "trino"],
+                "🌐 Networking": ["requests", "httpx", "socketio", "paramiko", "netmiko", "scapy", "dnspython", "asyncssh", "tqdm", "gevent", "websocket-client", "pyserial", "grpcio", "aiozmq", "uvloop", "aiofiles", "urllib3", "fastsocket", "twisted", "aiohttp-socks", "aioredis", "websockets"],
+                "🔒 Security": ["cryptography", "pyjwt", "requests-oauthlib", "paramiko", "pyOpenSSL", "passlib", "hashlib", "bcrypt", "python-nmap", "sqlmap", "scapy", "yara-python", "pwn", "mitmproxy", "themis", "pycryptodome", "tpm2-py", "pyspy", "volatility3", "pefile", "dpkt"],
+                 "🎨 GUI": ["tkinter", "pyqt5", "kivy", "wxpython", "pygame", "dearpygui", "pyglet", "flet", "toga", "eel", "qt-material", "pyside6", "pywebview", "guizero", "remif", "taipy"],
+                 "🧪 Testing": ["unittest", "pytest-cov", "hypothesis", "behave", "locust", "nose", "selenium", "playwright", "mock", "freezegun", "ddt", "tox", "robotframework", "hypothesis", "vcrpy", "responses", "faker", "factory-boy", "coveragepy", "parameterized"],
+                 "🧰 Utilities": ["click", "typer", "argparse", "rich", "tqdm", "colorama", "schedule", "python-dotenv", "shutil", "pathlib", "arrow", "toml", "json5", "xmltodict", "humanize", "pendulum", "inflect", "unidecode", "boltons", "delegator.py"],
+                "📚 Documentation": ["sphinx", "mkdocs", "pdoc", "readthedocs-sphinx-search", "numpydoc", "docutils", "recommonmark", "sphinx-rtd-theme", "furo", "m2r2", "autoapi", "plantuml", "pycco", "doc8", "griffe", "mistletoe", "marko"],
+                 "🎮 Game Dev": ["pygame", "pyglet", "arcade", "panda3d", "ursina", "cocos2d", "renpy", "pyopengl", "cymunk", "pytmx", "pybox2d", "pymunk", "moderngl", "glumpy", "glfw", "sdl2"],
+                "💹 Finance": ["numpy-financial", "pandas-datareader", "scikit-portfolio", "yfinance", "finmarketpy", "quantstats", "TA-Lib", "riskfolio-optimization", "alphalens", "pyfolio", "zipline", "backtrader", "quantlib-python", "pyfin", "quandl", "vnpy"],
+                 "🤖 Robotics": ["rospy", "pybullet", "mujoco-py", "open3d", "numpy-stl", "pyserial", "opencv-python", "pymavlink", "robotframework", "webots", "vpython", "transforms3d", "trimesh", "pyrealsense2", "aiortc", "gymnasium", "stable-baselines3"],
+                 "🧬 Bioinformatics": ["biopython", "scikit-bio", "pysam", "pandas-genomics", "statsmodels", "pybedtools", "matplotlib", "seaborn", "pyteomics", "networkx", "pyfaidx", "lifelines", "ngslib", "bcbio-nextgen", "scanpy", "anndata", "gseapy", "deepchem", "methylpy"],
+                 "🔭 Astronomy": ["astropy", "astroquery", "scikit-image", "photutils", "pyephem", "poliastro", "pyvista", "vtk", "healpy", "astroplan", "specutils", "sunpy", "gwpy", "lhorizon", "dustmaps", "reproject", "pymc3", "celerite"],
+                "🗺️ Geospatial": ["geopandas", "rasterio", "shapely", "pyproj", "fiona", "cartopy", "folium", "geopy", "earthpy", "descartes", "osmnx", "geohash", "momepy", "pyvista", "whitebox", "earthengine-api", "gisalchemist", "mapclassify", "spatialpandas"],
+                 "💡 IoT": ["pyserial", "paho-mqtt", "RPi.GPIO", "smbus2", "pysnmp", "can-utils", "zeroconf", "asyncio", "bluepy", "adafruit-circuitpython", "iotconnect", "thinger-io", "micropython", "homeassistant", "openhab", "tasmota", "esphome", "nodemcu"],
+                "💻 System": ["psutil", "subprocess32", "watchdog", "inotify", "pywin32", "sh", "delegator.py", "appdirs", "click", "typer", "pathlib", "colorama", "fire", "daemonize", "lsof", "ruamel.yaml", "py-cpuinfo", "platformdirs"]
+            }
+
+            selected_libraries = {}
+            for group, libraries in available_libraries.items():
+                 selected_libraries[group] = st.multiselect(f"{group}  ", libraries, default=[],key=f"lib_{group}")
+
+            selected_groups = st.multiselect("🗂️ Groups", list(available_libraries.keys()), default=[], help="Choose groups to include libraries.")
+
+
+        with st.expander("⚙️ Pip Settings"):
+            custom_requirements = st.text_input("➕ Custom Libraries:", placeholder="Specific libraries or versions (e.g., requests==2.28.1, numpy>=1.23.0)", help="List specific libraries and versions.")
+            specific_details = st.text_input("ℹ️ Specific Details", placeholder="Special pip install options, edge cases", help="Add specific details for the pip command generation.")
+            pip_options = st.text_input("⚙️ Pip Options", placeholder="Add custom pip options like --index-url or --no-cache", help="Add custom options to your pip command")
+            target_os = st.selectbox("🎯 Target OS", ["Linux 🐧", "Windows 🪟"], index=0, help="Choose the target Operating System.")
+
+
 
     with col2:
-        # User's base prompt
-        prompt_base = st.text_input("Describe the Puppet Bolt Plan:", placeholder="Ex: Execute a command to check disk space on all Linux servers, and save the output on /tmp/output.txt", key="prompt_base")
-        target_nodes_description = st.text_input("Target Nodes Description (text format, IP(s), Range or CIDR):", placeholder="Ex: all servers in the 192.168.0.0/24 network",key="target_nodes")
-
-        if prompt_presets != "None":
-            if prompt_base:
-               prompt_base = f"{prompt_presets} , {prompt_base}"
-            else:
-               prompt_base = prompt_presets;
-        if st.button("✨ Generate Puppet Bolt Plan"):
-            if not prompt_base:
-                st.error("⚠️ Please enter a plan description.")
+         # User's base prompt
+        prompt_base = st.text_input("📝 Describe the Pip Command:", placeholder="Ex: Install all libraries for web development and data analysis", key="prompt_base")
+        
+        if st.button("✨ Generate Pip Command"):
+            if not prompt_base and not selected_libraries and not selected_groups and not custom_requirements:
+                st.error("⚠️ Please enter a command description or select libraries/groups")
                 return
-            
-            if task_type != "command" and not task_code:
-                 st.error("⚠️ Please enter the code for the task.")
-                 return
-            
-            with st.spinner("⏳ Generating plan..."):
-                ai_code = generate_puppet_bolt_plan(
-                   prompt_base, 
-                   detail_level, 
-                   os_type,
-                   security_level,
+
+            selected_libs = []
+            for group in selected_groups:
+                 selected_libs.extend(selected_libraries[group])
+           
+            with st.spinner("⏳ Generating command..."):
+                ai_code = generate_pip_command(
+                   prompt_base,
                    model_name,
                    temperature,
                    top_p,
                    top_k,
                    max_tokens,
-                   prompt_detail,
-                   encoding,
-                   add_header,
-                   log_level,
-                   bolt_user,
-                   bolt_ssh_pass,
+                   selected_libs,
+                   selected_groups,
                    custom_requirements,
                    specific_details,
-                   target_nodes_description,
-                   task_type,
-                   task_code,
-                   parameters_description
+                   pip_options,
+                   target_os
                    )
 
                 if ai_code:
-                    st.markdown("### ✅ Generated Plan:")
-                    st.code(ai_code, language="yaml")
-
-                    short_title = prompt_base[:30].strip().replace(" ", "_").lower()
-                    file_name_yml, yml_code = parse_and_save_yml(ai_code, short_title, encoding, add_header)
-                    
-                    if file_name_yml and yml_code: #check if the file was generated
-                        st.download_button(
-                            label="⬇️ Download Plan (.yml)",
-                            data=yml_code,
-                            file_name=file_name_yml,
-                            mime="application/x-yaml",
-                        )
+                     st.markdown("### ✅ Generated Command:")
+                     st.code(ai_code, language="text") #to output all codes
+
+                     short_title = prompt_base[:30].strip().replace(" ", "_").lower()
+                     files = parse_and_save_code(ai_code, short_title)
+                     
+                     for key, value in files.items():
+                            st.download_button(
+                                label=f"⬇️ Download Command (.{(key)})",
+                                data=value["code"],
+                                file_name=value["name"],
+                                mime="text/plain",
+                            )
+
                 else:
-                    st.error("❌ Error generating the plan. Check the connection with the AI and try again.")
+                    st.error("❌ Error generating the command. Check the connection with the AI and try again.")
+        
 
 if __name__ == "__main__":
     main()