Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import os
 
+# --- 1. SETUP API CONNECTION ---
 API_KEY = os.getenv("HF_API_KEY")
 if not API_KEY:
     raise ValueError("⚠️ HF_API_KEY not found. Set it in your environment variables.")
@@ -9,6 +10,7 @@ if not API_KEY:
 # Use a small, free model that actually works
 client = InferenceClient("google/flan-t5-small", token=API_KEY)
 
+# --- 2. DEFINE THE LOGIC ---
 def analyze_biodiversity(species, habitat, region, activity, noise, flow, pollutants, breeding, disturbance):
     prompt = f"""
 You are BioVigilus, an expert AI Ecologist. Create a short, professional Impact Report with emojis.
@@ -28,27 +30,82 @@ OUTPUT REQUIRED:
 4. 🛡️ Mitigation (3 bullet points)
 5. ⚖️ Verdict
 """
-
     try:
         response = client.text_generation(prompt, max_new_tokens=300)
         return response[0].generated_text.strip()
     except Exception as e:
         return f"⚠️ Analysis Failed: {str(e)}"
 
-# Minimal Gradio UI
+# --- 3. CUSTOM STYLE ---
+custom_style = """
+<style>
+    body, .gradio-container { background-color: #0f172a !important; color: #e2e8f0 !important; }
+    h1 { color: #34d399 !important; text-align: center; font-size: 2.5em !important; margin-bottom: 0; }
+    h3 { color: #94a3b8 !important; text-align: center; margin-top: 0; }
+    
+    button.primary { 
+        background: linear-gradient(90deg, #10b981, #06b6d4) !important; 
+        border: none !important; 
+        color: black !important; 
+        font-weight: bold !important; 
+        font-size: 1.2em !important;
+    }
+    button.primary:hover { transform: scale(1.02); }
+
+    .gr-box, .gr-input, input, textarea { 
+        background-color: #1e293b !important; 
+        border: 1px solid #334155 !important; 
+        color: white !important;
+    }
+    label span { color: #64748b !important; font-weight: bold; }
+    
+    .output-text { font-family: monospace; background-color: #1e293b; padding: 15px; border-radius: 8px; border: 1px solid #475569; }
+</style>
+"""
+
+# --- 4. BUILD THE FULL UI ---
 with gr.Blocks() as app:
-    species = gr.Textbox("Species")
-    habitat = gr.Textbox("Habitat")
-    region = gr.Textbox("Region")
-    activity = gr.Textbox("Activity")
-    noise = gr.Slider(0,100)
-    flow = gr.Slider(0,100)
-    disturbance = gr.Slider(0,100)
-    pollutants = gr.Textbox("Pollutants")
-    breeding = gr.Radio(["Yes","No"], value="No")
-    output = gr.Markdown()
-    btn = gr.Button("Analyze")
-    btn.click(analyze_biodiversity, inputs=[species, habitat, region, activity, noise, flow, pollutants, breeding, disturbance], outputs=output)
+    
+    gr.HTML(custom_style)
+
+    with gr.Row():
+        gr.Markdown("# 🌿 BioVigilus")
+    gr.Markdown("### *AI-Powered Biodiversity Impact Analyzer*")
+    
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown("### 🦅 Species Info")
+            species = gr.Textbox(label="Species Name", placeholder="e.g. Snow Leopard")
+            habitat = gr.Textbox(label="Habitat", placeholder="e.g. Alpine Cliffs")
+            region = gr.Textbox(label="Region", placeholder="e.g. Gilgit-Baltistan")
+            activity = gr.Textbox(label="Activity", placeholder="e.g. Road Construction")
+            breeding = gr.Radio(["Yes", "No"], label="Breeding Season?", value="No")
+
+        with gr.Column():
+            gr.Markdown("### 📉 Stress Factors")
+            noise = gr.Slider(0, 100, label="Noise Level")
+            flow = gr.Slider(0, 100, label="Water Flow Change")
+            disturbance = gr.Slider(0, 100, label="Land Disturbance")
+            pollutants = gr.Textbox(label="Pollutants", placeholder="Dust, Oil, Chemicals")
+
+    btn = gr.Button("🔍 Analyze Ecological Impact", variant="primary")
+    
+    gr.Markdown("### 📋 Scientific Report")
+    output = gr.Markdown("Results will appear here...", elem_classes=["output-text"])
+
+    gr.Examples(
+        examples=[
+            ["Snow Leopard", "Rocky Mountains", "Gilgit", "Highway Construction", 85, 10, "Explosive Dust", "Yes", 70],
+            ["Indus Dolphin", "River Indus", "Sukkur", "Bridge Maintenance", 60, 80, "Oil Leak", "No", 20]
+        ],
+        inputs=[species, habitat, region, activity, noise, flow, pollutants, breeding, disturbance]
+    )
+
+    btn.click(
+        analyze_biodiversity, 
+        inputs=[species, habitat, region, activity, noise, flow, pollutants, breeding, disturbance], 
+        outputs=output
+    )
 
 if __name__ == "__main__":
     app.launch()