Update app.py

app.py

@@ -8,10 +8,6 @@ from intelligence_pipeline import (
 from safety_classifier import safety_check
 
 
-# ---------------------------------------------------------------------------
-# Mode Handlers
-# ---------------------------------------------------------------------------
-
 def handle_threat_intel(message, files):
     base = analyze_threat(message or "")
     if files:
@@ -46,14 +42,9 @@ MODE_ROUTER = {
 }
 
 
-# ---------------------------------------------------------------------------
-# Core message processor (with safety pre-check)
-# ---------------------------------------------------------------------------
-
 def process_message(message, files, history, mode):
     text = message or ""
 
-    # ---- SAFETY CHECK (blocks malicious or unsafe input) ----
     safety = safety_check(text)
     if safety["blocked"]:
         response = "⚠️ Safety Block: " + safety["reason"]
@@ -87,10 +78,6 @@ def download_chat(history):
     return tmp.name
 
 
-# ---------------------------------------------------------------------------
-# UI
-# ---------------------------------------------------------------------------
-
 with gr.Blocks(title="Threat Landscape Intelligence Console") as demo:
     gr.Markdown("## 🛰 APJ Threat Intelligence Console")
 
@@ -120,14 +107,12 @@ with gr.Blocks(title="Threat Landscape Intelligence Console") as demo:
         )
         send_btn = gr.Button("Send", variant="primary", scale=1)
 
-    # Button event
     send_btn.click(
         fn=process_message,
         inputs=[user_input, file_input, chat_state, mode],
         outputs=[chat, chat_state],
     ).then(lambda: "", None, user_input)
 
-    # Enter-key event
     user_input.submit(
         fn=process_message,
         inputs=[user_input, file_input, chat_state, mode],
@@ -137,5 +122,6 @@ with gr.Blocks(title="Threat Landscape Intelligence Console") as demo:
     clear_btn.click(fn=clear_chat, outputs=[chat, chat_state])
     download_btn.click(fn=download_chat, inputs=chat_state, outputs=download_file)
 
+
 if __name__ == "__main__":
     demo.launch()