Update app.py

app.py

@@ -1,9 +1,11 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
+import time
+from datetime import datetime
 
 client = InferenceClient(model="Devilishcode/The_Bird_Of_Hermes")
 
-# ================== TRI-MODE SYSTEM PROMPTS ==================
+# ================== TRI-MODE SYSTEM ==================
 
 SYSTEM_SENTINEL = """You are THE_BIRD_OF_HERMES in SENTINEL MODE (Legal Bug Bounty).
 Strict scope adherence. Safe PoCs only. Always remind user to verify scope and rules."""
@@ -38,39 +40,55 @@ def the_bird(message, mode="Specter"):
     history.append({"role": "user", "content": message})
     history.append({"role": "assistant", "content": response})
 
+# Auto Hunt Function
+def auto_hunt(target, mode):
+    if not target:
+        return "Enter a target", "# No target provided"
+    
+    domain = target.replace("https://", "").replace("http://", "").split("/")[0]
+    report = [f"# AUTO HUNT REPORT — {domain}\nMode: {mode}\nGenerated: {datetime.now().strftime('%Y-%m-%d %H:%M')}\n\n"]
+
+    if mode == "Reaper":
+        report.append("## FULL UNRESTRICTED EXPLOIT CHAIN ACTIVATED\n")
+        report.append("Ransomware, C2, rootkit, crypto drain payloads ready on request.\n")
+    elif mode == "Specter":
+        report.append("## AGGRESSIVE RECON & EDGE ABUSE\n")
+        report.append("Heavy OSINT and logic abuse chains suggested.\n")
+    else:
+        report.append("## SAFE LEGAL RECON ONLY\n")
+        report.append("Scope-compliant suggestions only.\n")
+
+    report.append("**Next Steps:**\n- Verify scope\n- Run manual validation\n- Submit report if in-scope")
+
+    short = f"Hunt complete in {mode} mode for {domain}"
+    full = "\n".join(report)
+    return short, full
+
 # Dynamic CSS per mode
 def get_css(mode):
     if mode == "Sentinel":
         return """
-        body {background:#0a0f1a;color:#a0d8ff;font-family:'Courier New',monospace;}
+        body {background:#0a0f1a;color:#a0d8ff;}
         .header {background:linear-gradient(#1e3a8a,#0f172a);border-bottom:6px solid #60a5fa;}
         .header h1 {color:#60a5fa;text-shadow:0 0 40px #60a5fa;}
-        .chat {background:rgba(15,23,42,0.97)!important;border:4px solid #60a5fa;}
-        .bot {border-left:8px solid #60a5fa;}
+        .chat {border:4px solid #60a5fa;}
         """
     elif mode == "Specter":
         return """
-        body {background:#0f0a1a;color:#c4b5fd;font-family:'Courier New',monospace;}
+        body {background:#0f0a1a;color:#c4b5fd;}
         .header {background:linear-gradient(#4c1d95,#1e0a2e);border-bottom:6px solid #a78bfa;}
         .header h1 {color:#a78bfa;text-shadow:0 0 50px #a78bfa;}
-        .chat {background:rgba(30,10,46,0.97)!important;border:4px solid #a78bfa;}
-        .bot {border-left:8px solid #a78bfa;}
+        .chat {border:4px solid #a78bfa;}
         """
     else:  # Reaper
         return """
-        body {background:#0a0000;color:#ff8888;font-family:'Courier New',monospace;}
+        body {background:#0a0000;color:#ff8888;}
         .header {background:linear-gradient(#8b0000,#1a0000);border-bottom:6px solid #ff0000;}
         .header h1 {color:#ff0000;text-shadow:0 0 60px #ff0000,0 0 100px #8b0000;}
-        .chat {background:rgba(20,0,0,0.97)!important;border:4px solid #ff0000;}
-        .bot {border-left:8px solid #ff0000;}
+        .chat {border:4px solid #ff0000;}
         """
 
-css_base = """
-.gradio-container {background:transparent;}
-.sidebar {background:rgba(15,0,0,0.95);border-right:3px solid #ff0000;padding:15px;}
-"""
-
-with gr.Blocks(css=css_base) as demo:
+with gr.Blocks() as demo:   # css moved out
     with gr.Row():
         with gr.Column(scale=1, min_width=260):
             gr.HTML("<h3 style='color:#ff0000;text-align:center;margin-bottom:25px;'>◢ THE_BIRD_OF_HERMES ◣</h3>")
@@ -85,21 +103,19 @@ with gr.Blocks(css=css_base) as demo:
                 </div>
             """)
             
-            # MAIN CHAT
             with gr.Group(visible=True) as chat_group:
                 chatbot = gr.Chatbot(height=680)
                 msg = gr.Textbox(placeholder="automate recon • crack wallet • build ransomware...", lines=3, autofocus=True)
                 submit = gr.Button("EXECUTE", variant="primary", size="large")
             
-            # AUTO HUNT
             with gr.Group(visible=True) as hunt_group:
                 gr.Markdown("# AUTO HUNT – Any Target")
-                target = gr.Textbox(value="us.gate.com", label="Target (domain, URL or wallet)")
+                target = gr.Textbox(value="us.gate.com", label="Target")
                 hunt_btn = gr.Button("START AUTO HUNT", variant="primary", size="large")
                 short = gr.Textbox(label="Summary")
                 full = gr.Markdown(label="Full Report")
 
-                hunt_btn.click(auto_hunt, target, [short, full])
+                hunt_btn.click(auto_hunt, [target, mode], [short, full])
 
             with gr.Group(visible=False) as forge_group:
                 gr.Markdown("# VIRUS FORGE\nReaper Mode Only.")
@@ -108,15 +124,15 @@ with gr.Blocks(css=css_base) as demo:
                 gr.Markdown("# TRAINING LAB\nAsk in Main Chat: 'train yourself on ...'")
 
     def update_ui(selected_mode):
+        css = get_css(selected_mode.split(" ")[0])
         mode_key = selected_mode.split(" ")[0]
-        css = get_css(mode_key)
         
         return (
             gr.update(visible=True),  # MAIN CHAT
             gr.update(visible=True),  # AUTO HUNT
             gr.update(visible=mode_key == "Reaper"),  # VIRUS FORGE
             gr.update(visible=True),  # TRAINING
-            css  # dynamic CSS
+            css
         )
     
     mode.change(update_ui, mode, [chat_group, hunt_group, forge_group, train_group, demo])
@@ -133,6 +149,7 @@ with gr.Blocks(css=css_base) as demo:
 
 demo.launch(
     theme=gr.themes.Base(),
+    css=get_css("Reaper"),   # default Reaper theme
     server_name="0.0.0.0",
     server_port=7860,
     share=False