Update app.py

app.py

@@ -1,150 +1,181 @@
 """
-ARMYINTEL 5D–MDMP Training Engine
-Purpose: Non‑kinetic military‑intelligence analysis simulator for education and defence training.
-Frameworks: 5D (Detect–Deter–Deny–Deliver–Defend) + MDMP
+ARMYINTEL 5D-MDMP INTELLIGENCE ANALYSIS ENGINE
+Purpose: Comprehensive multi-discipline intelligence synthesis using 5D & MDMP
+Domain: Military Intelligence Training / Defensive Operations
+Frameworks: 5D (Detect, Deter, Deny, Deliver, Destroy) + MDMP
 """
 
-import os, json, glob, concurrent.futures
+import os
+import json
 from datetime import datetime
 from docx import Document
+from docx.shared import Pt
 from docx.enum.text import WD_PARAGRAPH_ALIGNMENT
 import gradio as gr
 import openai
+import concurrent.futures
 
-# ---------- CONFIG ----------
+# -------------------------
+# CONFIG
+# -------------------------
 OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
 if not OPENAI_API_KEY:
-    raise ValueError("OPENAI_API_KEY missing in environment secrets.")
+    raise ValueError("OpenAI API Key not found in environment secrets.")
+
 openai.api_key = OPENAI_API_KEY
 
-INTEL_TECHNIQUES = [
-    "HUMINT","SIGINT","IMINT","OSINT","MASINT","COMINT","ELINT","TECHINT",
-    "FINT","SOCMINT","GEOINT","ALL‑SOURCE INTELLIGENCE","CRISIS INTELLIGENCE",
-    "COUNTERINTELLIGENCE","RED TEAMING","BLUE TEAMING",
-    "CTRIP Analysis","SALUT Analysis","Decision Trees",
-    "Bayesian Reasoning","Network Influence Mapping"
+MILITARY_INTEL_TECHNIQUES = [
+    "HUMINT", "SIGINT", "IMINT", "OSINT", "MASINT",
+    "COMINT", "ELINT", "TECHINT", "FINT", "SOCMINT",
+    "GEOINT", "ALL-SOURCE INTELLIGENCE", "CRISIS INTELLIGENCE",
+    "COUNTERINTELLIGENCE", "RED TEAMING", "BLUE TEAMING",
+    "CTRIP Analysis", "SALUT Analysis", "Decision Trees",
+    "Bayesian Reasoning", "Network Influence Mapping"
 ]
 
-# ---------- KNOWLEDGE‑BASE LOADING ----------
-def load_kb_snippets():
-    kb_text = ""
-    for f in glob.glob("knowledge_base/*"):
-        name = os.path.basename(f)
-        if any(k in name for k in ["5d","5D","mdmp","MDMP","jsonl","SOP","coa"]):
-            try:
-                with open(f, "r", encoding="utf‑8", errors="ignore") as fh:
-                    kb_text += f"\n--- {name} ---\n{fh.read()}\n"
-            except Exception:
-                pass
-    return kb_text[:20000]  # safety cap
-
-# ---------- CORE ANALYSIS ----------
-def analyze_with_technique(tech, problem, kb_text):
+# -------------------------
+# CORE 5D FUNCTION
+# -------------------------
+def apply_5d_to_technique(technique_name, problem_text):
     prompt = f"""
-Training context: defensive / educational only.
-
-Using the {tech} discipline, apply the 5D framework (Detect, Deter, Deny, Deliver, Defend)
-and MDMP reasoning to analyse the following situation:
-
-{problem}
-
-Reference knowledge (for orientation only):
-{kb_text}
-
-Return JSON:
-{{
- "technique": "{tech}",
- "findings": "...",
- "5D": {{
-   "Detect": "...", "Deter": "...", "Deny": "...", "Deliver": "...", "Defend": "..."
- }},
- "recommended_actions": ["...", "..."],
- "confidence": {{"level":"High|Medium|Low","reason":"..."}}
-}}
+Conduct a professional military intelligence analysis for the following operational or training situation:
+Technique: {technique_name}
+Apply the 5D framework (Detect, Deter, Deny, Deliver, Destroy) and give structured findings.
+Include likely enemy capabilities, indicators, vulnerabilities, and recommended defensive measures.
+Problem: {problem_text}
 """
     try:
-        resp = openai.ChatCompletion.create(
+        response = openai.ChatCompletion.create(
             model="gpt-4-turbo",
             messages=[
-                {"role":"system","content":"You are a senior military intelligence instructor. Provide non‑kinetic, training‑safe analysis."},
-                {"role":"user","content":prompt}
+                {"role": "system", "content": "You are a senior army intelligence officer producing analytical reports for training and defensive planning."},
+                {"role": "user", "content": prompt}
             ],
             temperature=0.5,
             max_tokens=600
         )
-        txt = resp["choices"][0]["message"]["content"].strip()
+        text = response['choices'][0]['message']['content'].strip()
         try:
-            data = json.loads(txt)
+            data = json.loads(text)
         except:
-            data = {"technique":tech,"findings":txt,"5D":{},"recommended_actions":[],
-                    "confidence":{"level":"Unknown","reason":"Text only"}}
+            data = {
+                "technique": technique_name,
+                "findings": text,
+                "5D": {},
+                "recommended_actions": [],
+                "confidence": {"level":"Unknown","reason":"Model returned narrative text"}
+            }
+        data.setdefault("technique", technique_name)
+        data.setdefault("findings", "")
+        data.setdefault("5D", {})
+        data.setdefault("recommended_actions", [])
+        data.setdefault("confidence", {"level":"Unknown","reason":"Missing"})
         return data
     except Exception as e:
-        return {"technique":tech,"findings":f"Analysis failed: {e}","5D":{},
-                "recommended_actions":[],"confidence":{"level":"Error","reason":str(e)}}
-
-# ---------- DOCX REPORT ----------
-def build_docx(analyses):
+        return {
+            "technique": technique_name,
+            "findings": f"Analysis incomplete: {str(e)}",
+            "5D": {},
+            "recommended_actions": [],
+            "confidence": {"level":"Error","reason": str(e)}
+        }
+
+# -------------------------
+# DOCX SAVE
+# -------------------------
+def save_docx(report_text, analyses):
     os.makedirs("reports", exist_ok=True)
-    fname = f"reports/ArmyIntel_5D_MDMP_Report_{datetime.now().strftime('%Y%m%d_%H%M%S')}.docx"
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    filename = f"reports/ArmyIntel_5D_MDMP_Report_{timestamp}.docx"
+
     doc = Document()
-    h = doc.add_heading("INTELLIGENCE ANALYSIS – INTELLIGENCE WARFARE 5D MDMP TRAINING REPORT",0)
-    h.alignment = WD_PARAGRAPH_ALIGNMENT.CENTER
-    doc.add_paragraph(f"Generated on {datetime.now():%d %b %Y %H:%M:%S}", style="Caption")
-    doc.add_paragraph("—"*30)
-
-    for i,a in enumerate(analyses,1):
-        doc.add_heading(f"{i}. {a.get('technique','Unknown')}", level=1)
-        doc.add_paragraph(a.get("findings","No findings."))
-
-        doc.add_heading("5D Analysis", level=2)
-        for d in ["Detect","Deter","Deny","Deliver","Defend"]:
-            doc.add_paragraph(f"• {d}: {a.get('5D',{}).get(d,'')}", style="List Bullet")
-
-        doc.add_heading("Recommended Actions", level=2)
-        if a.get("recommended_actions"):
-            for r in a["recommended_actions"]:
-                doc.add_paragraph(f"• {r}", style="List Bullet")
+
+    # Title Page
+    title = doc.add_heading("ARMYINTEL 5D–MDMP INTELLIGENCE REPORT", 0)
+    title.alignment = WD_PARAGRAPH_ALIGNMENT.CENTER
+    doc.add_paragraph("Directorate of Military Intelligence (Training Use Only)", style='Intense Quote')
+    doc.add_paragraph(f"Report generated on: {datetime.now().strftime('%d %b %Y %H:%M:%S')}", style='Caption')
+    doc.add_paragraph("\n---\n", style='Normal')
+
+    # Individual Technique Analyses
+    for i, a in enumerate(analyses, 1):
+        doc.add_heading(f"{i}. {a.get('technique','Unknown')}", level=1)
+        doc.add_paragraph("Findings:", style='Heading 3')
+        doc.add_paragraph(a.get('findings',''), style='Normal')
+
+        doc.add_paragraph("5D Analysis:", style='Heading 3')
+        five_d = a.get('5D', {})
+        for d_key in ["Detect","Deter","Deny","Deliver","Destroy"]:
+            doc.add_paragraph(f"- {d_key}: {five_d.get(d_key,'')}", style='List Bullet')
+
+        doc.add_paragraph("Recommended Actions:", style='Heading 3')
+        if a.get('recommended_actions'):
+            for rec in a.get('recommended_actions', []):
+                doc.add_paragraph(f"- {rec}", style='List Bullet')
         else:
-            doc.add_paragraph("• None / pending analysis")
-
-        c = a.get("confidence",{})
-        doc.add_paragraph(f"Confidence: {c.get('level','Unknown')} – {c.get('reason','')}")
-        doc.add_paragraph(" ")
-
-    doc.add_heading("MDMP Summary", level=1)
-    doc.add_paragraph(
-        "1. Problem identification → 2. Intelligence gathering → 3. COA development → "
-        "4. COA analysis → 5. Decision & execution (Training Simulation only)."
-    )
-    doc.save(fname)
-    return fname
-
-# ---------- MAIN ENGINE ----------
-def run_full_analysis(problem):
-    kb_text = load_kb_snippets()
-    analyses=[]
-    with concurrent.futures.ThreadPoolExecutor(max_workers=6) as pool:
-        futures={pool.submit(analyze_with_technique,t,problem,kb_text):t for t in INTEL_TECHNIQUES}
-        for f in concurrent.futures.as_completed(futures):
-            analyses.append(f.result())
-    return build_docx(analyses)
-
-# ---------- GRADIO UI ----------
-def start_analysis(problem):
-    if not problem.strip():
-        problem="Training scenario: suspected hostile surveillance around forward base; assess defensive posture."
-    path=run_full_analysis(problem)
-    return path
-
-demo=gr.Interface(
-    fn=start_analysis,
-    inputs=gr.Textbox(label="Enter Training / Defensive Problem", lines=8),
-    outputs=gr.File(label="Download Training Intelligence Report"),
-    title="ARMYINTEL 5D + MDMP Training Engine",
-    description="Educational simulation engine for non‑kinetic intelligence analysis using 21 techniques + 5D + MDMP."
+            doc.add_paragraph("- None provided / pending analysis", style='List Bullet')
+
+        confidence = a.get('confidence', {})
+        doc.add_paragraph(f"Confidence Level: {confidence.get('level','Unknown')}", style='Normal')
+        doc.add_paragraph(f"Reason: {confidence.get('reason','')}", style='Normal')
+        doc.add_paragraph("\n", style='Normal')
+
+    # MDMP Summary
+    doc.add_heading("MDMP Summary", level=1)
+    mdmp_summary = f"""
+MDMP Step Summary:
+1. Intelligence requirements derived from all {len(MILITARY_INTEL_TECHNIQUES)} disciplines.
+2. Threat courses of action evaluated using 5D framework.
+3. Red / Blue Team comparisons generated for decision support.
+4. Recommended defensive COAs prioritized for implementation.
+"""
+    doc.add_paragraph(mdmp_summary, style='Normal')
+
+    doc.save(filename)
+    return filename
+
+# -------------------------
+# FULL ANALYSIS ENGINE
+# -------------------------
+def run_full_analysis(problem_text):
+    analyses = []
+    with concurrent.futures.ThreadPoolExecutor(max_workers=7) as executor:
+        futures = {executor.submit(apply_5d_to_technique, tech, problem_text): tech for tech in MILITARY_INTEL_TECHNIQUES}
+        for future in concurrent.futures.as_completed(futures):
+            analyses.append(future.result())
+
+    fused = "\n\n".join([
+        f"{i+1}. {a.get('technique','Unknown')}\nFindings: {a.get('findings','')}"
+        for i, a in enumerate(analyses)
+    ])
+
+    mdmp = f"""
+MDMP Step Summary:
+1. Intelligence requirements synthesized across all {len(MILITARY_INTEL_TECHNIQUES)} methods.
+2. Defensive COAs refined via 5D reasoning.
+3. Enemy indicators and vulnerabilities mapped.
+4. Final recommendations structured for command review.
+"""
+    report_text = f"=== Combined Analysis ===\n{fused}\n\n=== MDMP Summary ===\n{mdmp}"
+    return report_text, analyses, fused, mdmp
+
+# -------------------------
+# GRADIO UI
+# -------------------------
+def analyze_problem(problem_text):
+    if not problem_text.strip():
+        problem_text = "Default scenario: enemy infiltration risk in forward area; develop full intelligence picture."
+    report_text, analyses, fused, mdmp = run_full_analysis(problem_text)
+    filename = save_docx(report_text, analyses)
+    return filename
+
+demo = gr.Interface(
+    fn=analyze_problem,
+    inputs=gr.Textbox(label="Enter Operational Problem or Training Scenario", lines=10),
+    outputs=gr.File(label="Download Intelligence Report"),
+    title="ARMYINTEL 5D–MDMP Intelligence Engine",
+    description="Enter any enemy or threat-related situation to generate a structured military intelligence report using 21 collection techniques and the 5D + MDMP frameworks."
 )
 
-if __name__=="__main__":
-    demo.launch(share=True)
-
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=7860, share=True)