Upload 4 files

.gitattributes

@@ -1,35 +1,35 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,111 @@
+import json
+import gradio as gr
+
+from orchestrator import run_tara_analysis
+
+
+# -----------------------------------
+# Main Analysis Function
+# -----------------------------------
+def analyze_trademark(report_json):
+
+    try:
+
+        if isinstance(report_json, str):
+            report_json = json.loads(report_json)
+
+        result = run_tara_analysis(report_json)
+
+        return {
+            "status": "success",
+            "tara_analysis": result
+        }
+
+    except Exception as e:
+
+        return {
+            "status": "error",
+            "message": str(e)
+        }
+
+# -----------------------------------
+# Example Input
+# -----------------------------------
+
+example_json = {
+    "application_overview": {
+        "serial_number": "97654321",
+        "mark_text": "NOVA",
+        "applicant_name": "Nova Corp LLC",
+        "filing_date": "2024-03-15",
+        "filing_basis": "1(b)",
+        "classes": [9, 42],
+        "identification": "Computer software for..."
+    },
+    "overall_assessment": {
+        "risk_level": "LOW",
+        "confidence_score": 0.80
+    },
+    "classification_analysis": {
+        "claimed_class": 9,
+        "suggested_class": 42,
+        "status": "Valid",
+        "confidence": "High"
+    },
+    "conflict_analysis": {
+        "total_conflicts": 2,
+        "high_risk_conflicts": 1,
+        "sample_conflicts": [
+            {
+                "conflicting_mark": "NOVA TECH",
+                "serial": "88123456",
+                "risk": "HIGH",
+                "similarity_score": 0.87
+            }
+        ]
+    },
+    "distinctiveness_analysis": {
+        "descriptive_score": 0.42,
+        "generic_score": 0.18
+    },
+    "validation_issues": []
+}
+
+
+# -----------------------------------
+# Gradio UI
+# -----------------------------------
+
+with gr.Blocks(title="TARA AI Trademark Analyzer") as app:
+
+    gr.Markdown("# TARA AI — Trademark Risk Analyzer")
+
+    gr.Markdown(
+        "Paste the structured trademark report JSON and run the analysis pipeline."
+    )
+
+    input_box = gr.Textbox(
+        label="Trademark Report JSON",
+        value=json.dumps(example_json, indent=2),
+        lines=20
+    )
+
+    run_button = gr.Button("Run TARA Analysis")
+
+    output_box = gr.JSON(
+        label="TARA AI Analysis Output"
+    )
+
+    run_button.click(
+        fn=analyze_trademark,
+        inputs=input_box,
+        outputs=output_box,
+        api_name="analyze_trademark"
+    )
+
+
+# -----------------------------------
+# Launch
+# -----------------------------------
+
+app.launch(server_name="0.0.0.0", server_port=7860, ssr_mode=False)

orchestrator.py

@@ -0,0 +1,187 @@
+import re
+import json
+import time
+
+from agents.risk_analyzer import run_risk_analyzer
+from agents.refusal_predictor import run_refusal_predictor
+from agents.strategy_planner import run_strategy_planner
+from agents.fix_generator import run_fix_generator
+
+
+MAX_RETRIES = 2
+
+
+# ---------------------------------
+# JSON Parsing Utilities
+# ---------------------------------
+
+def safe_json_parse(text):
+    """
+    Try strict JSON parse first.
+    If it fails, try extracting JSON block.
+    If still fails, return raw text.
+    """
+
+    try:
+        return json.loads(text)
+    except Exception:
+        pass
+
+    try:
+        match = re.search(r"\{.*\}", text, re.DOTALL)
+        if match:
+            return json.loads(match.group())
+    except Exception:
+        pass
+
+    return text
+# ---------------------------------
+# Safe Key Extractor
+# ---------------------------------
+
+def safe_get(data, key):
+    """
+    Safely extract a key from agent output.
+    If output is not dict or key missing,
+    return the raw data so pipeline continues.
+    """
+
+    if isinstance(data, dict):
+        return data.get(key, data)
+
+    return data
+
+
+
+# ---------------------------------
+# Agent Execution with Retry
+# ---------------------------------
+
+def run_agent_with_retry(agent_fn, payload, expected_key):
+
+    for attempt in range(MAX_RETRIES + 1):
+
+        try:
+            result = agent_fn(payload)
+
+            parsed = safe_json_parse(result)
+
+            if isinstance(parsed, dict) and expected_key in parsed:
+                return parsed
+            if attempt == MAX_RETRIES:
+                return parsed
+
+        except Exception as e:
+            print(f"Agent error: {e}")
+
+        if attempt < MAX_RETRIES:
+            time.sleep(1)
+
+    return result
+
+
+# ---------------------------------
+# Agent Pipeline
+# ---------------------------------
+
+def run_pipeline(report_json):
+
+    print("Starting TARA agent pipeline")
+
+    # --------------------
+    # Agent 1 — Risk Analyzer
+    # --------------------
+
+    risk_payload = {
+        "report_json": report_json
+    }
+
+    risk_output = run_agent_with_retry(
+        run_risk_analyzer,
+        risk_payload,
+        "identified_risks"
+    )
+
+    print("Agent 1 completed")
+
+
+    # --------------------
+    # Agent 2 — Refusal Predictor
+    # --------------------
+
+    refusal_payload = {
+        "report_json": report_json,
+        "identified_risks": safe_get(risk_output, "identified_risks")
+    }
+
+    refusal_output = run_agent_with_retry(
+        run_refusal_predictor,
+        refusal_payload,
+        "predicted_refusals"
+    )
+
+    print("Agent 2 completed")
+
+
+    # --------------------
+    # Agent 3 — Strategy Planner
+    # --------------------
+
+    strategy_payload = {
+        "report_json": report_json,
+        "identified_risks": safe_get(risk_output, "identified_risks"),
+        "predicted_refusals": safe_get(refusal_output, "predicted_refusals")
+    }
+
+    strategy_output = run_agent_with_retry(
+        run_strategy_planner,
+        strategy_payload,
+        "strategy_plan"
+    )
+
+    print("Agent 3 completed")
+
+
+    # --------------------
+    # Agent 4 — Fix Generator
+    # --------------------
+
+    fix_payload = {
+        "report_json": report_json,
+        "identified_risks": safe_get(risk_output, "identified_risks"),
+        "predicted_refusals": safe_get(refusal_output, "predicted_refusals"),
+        "strategy_plan": safe_get(strategy_output, "strategy_plan")
+    }
+
+    fix_output = run_agent_with_retry(
+        run_fix_generator,
+        fix_payload,
+        "application_improvements"
+    )
+
+    print("Agent 4 completed")
+
+
+    return {
+         "agent1_output": risk_output,
+         "agent2_output": refusal_output,
+         "agent3_output": strategy_output,
+         "agent4_output": fix_output
+    }
+
+
+# ---------------------------------
+# Public API Function
+# ---------------------------------
+
+def run_tara_analysis(report_json):
+
+    if not isinstance(report_json, dict):
+        raise ValueError("report_json must be a dictionary")
+
+    result = run_pipeline(report_json)
+
+    return {
+        "status": "success",
+        "tara_analysis": result
+    }

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+groq
+python-dotenv