ScottzillaSystems commited on
Commit
cc8635b
·
verified ·
1 Parent(s): a462c2f

Fix: move CSS to global scope to resolve NameError in main()

Browse files
Files changed (1) hide show
  1. app.py +5 -6
app.py CHANGED
@@ -660,12 +660,11 @@ class ChatInterface:
660
  # Gradio UI
661
  # ============================================================
662
 
663
- def create_ui(config: AgentZeroConfig, chat_interface: ChatInterface, defense: PromptInjectionDefense):
664
- css = """
665
- .agent-zero-header { background: linear-gradient(135deg, #0a0a0a 0%, #1a1a2e 50%, #16213e 100%); padding: 20px; border-radius: 10px; border: 1px solid #00ff41; margin-bottom: 20px; }
666
  .agent-zero-header h1 { color: #00ff41; font-family: 'Courier New', monospace; text-shadow: 0 0 10px rgba(0,255,65,0.5); }
667
- footer { visibility: hidden; }
668
- """
 
669
  with gr.Blocks(title="Pentesting Agent Zero") as demo:
670
  gr.HTML("""<div class="agent-zero-header">
671
  <h1>🛡️ PENTESTING AGENT ZERO</h1>
@@ -727,7 +726,7 @@ def main():
727
  print(f"[+] Defense: {'ACTIVE' if config.prompt_injection_defense else 'DISABLED'}")
728
  print(f"[+] Model: {config.agent_model}\n")
729
  demo = create_ui(config, chat_interface, defense)
730
- demo.launch(css=css, server_name="0.0.0.0", server_port=7860, mcp_server=config.mcp_enabled, share=False, theme=gr.themes.Monochrome(primary_hue="green", secondary_hue="gray"))
731
 
732
  if __name__ == "__main__":
733
  main()
 
660
  # Gradio UI
661
  # ============================================================
662
 
663
+ CSS_STYLE = """ .agent-zero-header { background: linear-gradient(135deg, #0a0a0a 0%, #1a1a2e 50%, #16213e 100%); padding: 20px; border-radius: 10px; border: 1px solid #00ff41; margin-bottom: 20px; }
 
 
664
  .agent-zero-header h1 { color: #00ff41; font-family: 'Courier New', monospace; text-shadow: 0 0 10px rgba(0,255,65,0.5); }
665
+ footer { visibility: hidden; }"""
666
+
667
+ def create_ui(config: AgentZeroConfig, chat_interface: ChatInterface, defense: PromptInjectionDefense):
668
  with gr.Blocks(title="Pentesting Agent Zero") as demo:
669
  gr.HTML("""<div class="agent-zero-header">
670
  <h1>🛡️ PENTESTING AGENT ZERO</h1>
 
726
  print(f"[+] Defense: {'ACTIVE' if config.prompt_injection_defense else 'DISABLED'}")
727
  print(f"[+] Model: {config.agent_model}\n")
728
  demo = create_ui(config, chat_interface, defense)
729
+ demo.launch(css=CSS_STYLE, server_name="0.0.0.0", server_port=7860, mcp_server=config.mcp_enabled, share=False, theme=gr.themes.Monochrome(primary_hue="green", secondary_hue="gray"))
730
 
731
  if __name__ == "__main__":
732
  main()