Fix: move CSS to global scope to resolve NameError in main()
Browse files
app.py
CHANGED
|
@@ -660,12 +660,11 @@ class ChatInterface:
|
|
| 660 |
# Gradio UI
|
| 661 |
# ============================================================
|
| 662 |
|
| 663 |
-
|
| 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=
|
| 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()
|