Upload 3 files

README.md

@@ -1,13 +1,27 @@
----
-title: Justiceai Analyzer
-emoji: 😻
-colorFrom: blue
-colorTo: indigo
-sdk: gradio
-sdk_version: 5.44.1
-app_file: app.py
-pinned: false
-license: cc-by-4.0
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# JusticeAI – Criminal Record Analyzer
+
+JusticeAI is an AI-powered legal assistant designed to analyze and summarize criminal case records. It extracts key information such as names, crimes, locations, and judgments, then provides legal insights including applicable laws and recommended punishments based on Nigerian criminal law.
+
+## What It Does
+
+- Accepts pasted or uploaded criminal case text
+- Automatically summarizes long legal reports
+- Extracts named entities such as people, dates, crimes, courts, and locations
+- Classifies crime type based on Nigerian law
+- Recommends appropriate punishment according to legal sections
+
+## Technologies Used
+
+- Python
+- Hugging Face Transformers (`transformers`, `pipeline`)
+- PyTorch
+- Gradio (UI interface)
+
+## Getting Started
+
+To run locally:
+
+1. Clone the repository or upload files to your Hugging Face Space.
+2. Install dependencies:
+   ```bash
+   pip install -r requirements.txt

main.py

@@ -0,0 +1,80 @@
+from transformers import pipeline
+import gradio as gr
+
+# Load pre-trained pipelines
+summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
+ner = pipeline("ner", model="Davlan/bert-base-multilingual-cased-ner-hrl", aggregation_strategy="simple")
+
+# Basic crime-to-punishment mapping (based on Nigerian laws)
+crime_punishment_map = {
+    "theft": {
+        "law": "Criminal Code Act, Section 390",
+        "punishment": "Up to 3 years imprisonment"
+    },
+    "armed robbery": {
+        "law": "Robbery and Firearms Act, Section 1",
+        "punishment": "Death penalty or life imprisonment"
+    },
+    "internet fraud": {
+        "law": "Cybercrime Act, 2015",
+        "punishment": "Minimum of 7 years imprisonment"
+    },
+    "rape": {
+        "law": "Criminal Law of Lagos State, Section 260",
+        "punishment": "Life imprisonment"
+    },
+    "murder": {
+        "law": "Criminal Code Act, Section 319",
+        "punishment": "Death penalty"
+    },
+    "assault": {
+        "law": "Criminal Code Act, Section 351",
+        "punishment": "1 year imprisonment"
+    }
+}
+
+# Basic keyword-based classification
+def classify_crime(text):
+    text = text.lower()
+    for crime in crime_punishment_map:
+        if crime in text:
+            return crime, crime_punishment_map[crime]
+    return "unknown", {
+        "law": "N/A",
+        "punishment": "No specific punishment found. Manual review required."
+    }
+
+# Main function
+def analyze_text(text):
+    # Step 1: Run summarization
+    summary = summarizer(text, max_length=80, min_length=30, do_sample=False)[0]["summary_text"]
+
+    # Step 2: Named Entity Recognition
+    entities = ner(text)
+
+    # Step 3: Crime classification and punishment recommendation
+    crime_type, recommendation = classify_crime(text)
+
+    # Final output
+    return {
+        "Summary": summary,
+        "Extracted Entities": entities,
+        "Legal Analysis": {
+            "Crime Type": crime_type.title() if crime_type != "unknown" else "Unknown",
+            "Applicable Law": recommendation["law"],
+            "Recommended Punishment": recommendation["punishment"]
+        }
+    }
+
+# Gradio Interface
+gr.Interface(
+    fn=analyze_text,
+    inputs=gr.Textbox(lines=12, label="Paste Criminal Case Text"),
+    outputs=[
+        gr.Textbox(label="Summary"),
+        gr.JSON(label="Extracted Entities"),
+        gr.JSON(label="Legal Analysis / Recommended Punishment")
+    ],
+    title="JusticeAI - Legal Case Analyzer",
+    description="Paste any criminal case report. This AI will summarize it, extract important entities, and recommend the legal punishment based on Nigerian law."
+).launch()

requirements.txt

@@ -0,0 +1,3 @@
+transformers
+torch
+gradio