Create app.py

app.py

@@ -0,0 +1,94 @@
+import os
+import gradio as gr
+from groq import Groq
+
+# Safely pull the API key from Hugging Face Space Settings -> Variables and secrets
+GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
+
+def generate_network_script(user_goal, library, device_type):
+    # Check if the user forgot to add the API key in settings
+    if not GROQ_API_KEY:
+        return "# Error \nGROQ_API_KEY is missing. Please add it in your Space Settings -> Variables and secrets."
+
+    client = Groq(api_key=GROQ_API_KEY)
+    
+    # System Prompt telling AI to act as an expert and output ONLY code
+    system_prompt = """You are a Senior Network Automation Architect. 
+    Your task is to generate production-ready Python automation scripts. 
+    Output ONLY the raw Python code. Do not include markdown formatting like ```python, and do not include any conversational text before or after the code."""
+    
+    user_prompt = f"Write a Python script using the '{library}' library for '{device_type}' devices to accomplish the following: {user_goal}"
+    
+    try:
+        chat_completion = client.chat.completions.create(
+            messages=[
+                {"role": "system", "content": system_prompt},
+                {"role": "user", "content": user_prompt}
+            ],
+            model="llama-3.3-70b-versatile",
+            temperature=0.1, # Low temperature for accurate coding
+        )
+        return chat_completion.choices[0].message.content
+    except Exception as e:
+        return f"# Error \n{str(e)}"
+
+# Custom CSS for a premium SaaS look
+custom_css = """
+.gradio-container {font-family: 'Inter', sans-serif;}
+.header-box {text-align: center; padding: 20px; background: linear-gradient(90deg, #1e3a8a, #3b82f6); color: white; border-radius: 10px; margin-bottom: 20px;}
+.header-box h1 {margin: 0; font-size: 28px; font-weight: bold; color: white !important;}
+.header-box p {margin: 5px 0 0 0; opacity: 0.9;}
+"""
+
+# Build the Gradio UI
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="indigo"), css=custom_css) as app:
+    
+    # Custom HTML Header
+    gr.HTML("""
+        <div class="header-box">
+            <h1>⚙️ AutoNet Architect</h1>
+            <p>AI-Powered Python Script Generator for Network Engineers</p>
+        </div>
+    """)
+    
+    with gr.Row():
+        # Left side inputs
+        with gr.Column(scale=1):
+            with gr.Group():
+                library = gr.Dropdown(
+                    choices=["Netmiko", "NAPALM", "Nornir"], 
+                    value="Netmiko", 
+                    label="1. Select Python Library",
+                    info="Which automation framework do you want to use?"
+                )
+                device_type = gr.Dropdown(
+                    choices=["cisco_ios", "cisco_nxos", "arista_eos", "juniper_junos"], 
+                    value="cisco_ios", 
+                    label="2. Select Device OS",
+                    info="What type of devices are you connecting to?"
+                )
+                user_goal = gr.Textbox(
+                    lines=5, 
+                    label="3. Describe the Task", 
+                    placeholder="e.g., Log into 50 switches, backup the running-config, and save them as text files with the hostname and date."
+                )
+            
+            generate_btn = gr.Button("✨ Generate Code", variant="primary", size="lg")
+            
+        # Right side output
+        with gr.Column(scale=2):
+            output_code = gr.Code(
+                label="Production-Ready Python Script",
+                language="python",
+                interactive=False
+            )
+            
+    # Connect UI to logic
+    generate_btn.click(
+        fn=generate_network_script, 
+        inputs=[user_goal, library, device_type], 
+        outputs=output_code
+    )
+
+# Launch app (Hugging Face handles the web server routing automatically)
+app.launch()