Update app.py

app.py

@@ -34,7 +34,7 @@ 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_ansible_playbook(prompt_base, detail_level, os_type, security_level, model_name, temperature, top_p, top_k, max_tokens, prompt_detail, encoding, add_header, log_level, ansible_user, ansible_ssh_pass, ansible_become, custom_requirements, specific_details, target_hosts):
+def generate_ansible_playbook(prompt_base, detail_level, os_type, security_level, model_name, temperature, top_p, top_k, max_tokens, prompt_detail, encoding, add_header, log_level, ansible_user, ansible_ssh_pass, ansible_become, custom_requirements, specific_details, target_hosts_description):
     """Generates an Ansible playbook based on user settings."""
     prompt = f"""
     You are an expert Ansible automation engineer. Your task is to generate a complete, secure, and efficient Ansible playbook based on the following description:
@@ -47,7 +47,7 @@ def generate_ansible_playbook(prompt_base, detail_level, os_type, security_level
     **Security Level:** {security_level}
     **Prompt Detail Level**:{prompt_detail}
     **Operating System:** {os_type}
-    **Target Hosts:** {target_hosts if target_hosts else "None"}
+    **Target Hosts:** {target_hosts_description if target_hosts_description else "None"}
     **Ansible User:** {ansible_user if ansible_user else "root"}
     **Ansible SSH Password:** {ansible_ssh_pass if ansible_ssh_pass else "None"}
      **Ansible Become:** {ansible_become}
@@ -60,6 +60,7 @@ def generate_ansible_playbook(prompt_base, detail_level, os_type, security_level
     - 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 `hosts` section of the playbook, must include the list of the target hosts in the description, if the target hosts are not specified you must create your own hosts list, please avoid using "all" if the target hosts are specified.
     - Explore different approaches, techniques, and advanced practices, always prioritizing security and efficiency.
     - Use advanced Ansible resources such as modules, roles, variables, functions, conditionals, loops, and handlers, when necessary.
     - Unless the user specifies otherwise, use the most current and secure versions of the modules, using ansible and following best practices.
@@ -108,42 +109,6 @@ def parse_and_save_yml(ai_code, short_title, encoding, add_header):
         
     return file_name, yml_code
 
-def validate_ip_input(ip_input):
-    """Validates and parses IP address input."""
-    try:
-        if "," in ip_input:
-            ips = [ip.strip() for ip in ip_input.split(",")]
-            for ip in ips:
-                 if "/" in ip:
-                    ipaddress.ip_network(ip, strict=False) #verify if is a network
-                 elif "-" in ip:
-                      parts = ip.split("-")
-                      if len(parts) != 2:
-                          return False, "Invalid Range Format"
-                      
-                      ipaddress.ip_address(parts[0])
-                      ipaddress.ip_address(parts[1])
-                 else:
-                     ipaddress.ip_address(ip)  #verify if is a valid address
-            return True, ips
-        elif "/" in ip_input:
-              ipaddress.ip_network(ip_input, strict=False) #verify if is a network
-              return True, [ip_input]
-        elif "-" in ip_input:
-              parts = ip_input.split("-")
-              if len(parts) != 2:
-                  return False, "Invalid Range Format"
-              
-              ipaddress.ip_address(parts[0])
-              ipaddress.ip_address(parts[1])
-              return True,[ip_input]
-        else:
-            ipaddress.ip_address(ip_input)
-            return True, [ip_input]
-    except ValueError as e:
-        return False, str(e)
-    except Exception as e:
-        return False, str(e)
 
 def main():
     st.title("⚙️ Gemini2 Ansible Gen Pro by [Elias Andrade](https://github.com/chaos4455)")
@@ -193,7 +158,7 @@ def main():
     with col2:
         # User's base prompt
         prompt_base = st.text_input("Describe the Ansible Playbook:", placeholder="Ex: Install apache2 and configure a default page for all the servers", key="prompt_base")
-        target_hosts = st.text_input("Target Hosts (IP(s), Range or CIDR, separated by comma):", placeholder="192.168.0.1,192.168.0.2,192.168.0.1/24 or 192.168.0.1-255",key="target_hosts")
+        target_hosts_description = st.text_input("Target Hosts Description (text format, IP(s), Range or CIDR):", placeholder="Ex: all the servers in 192.168.0.0/24 network",key="target_hosts")
 
         if prompt_presets != "None":
             if prompt_base:
@@ -204,11 +169,6 @@ def main():
             if not prompt_base:
                 st.error("⚠️ Please enter a playbook description.")
                 return
-            
-            valid_ip, parsed_ips = validate_ip_input(target_hosts)
-            if not valid_ip:
-               st.error(f"⚠️ Invalid IP Address or Range: {parsed_ips}")
-               return
 
             with st.spinner("⏳ Generating playbook..."):
                 ai_code = generate_ansible_playbook(
@@ -230,7 +190,7 @@ def main():
                    ansible_become,
                    custom_requirements,
                    specific_details,
-                   parsed_ips if parsed_ips else None
+                   target_hosts_description # Pass only the description
                    )
 
                 if ai_code: