Update app.py

app.py

@@ -41,20 +41,14 @@ def split_into_paragraphs(text: str):
     return paragraphs
 
 def analyze_text_block(text: str):
-    """Analyze a single paragraph and return AI/Human probability."""
+    """Analyze a block of text and return AI/Human probabilities."""
     inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
     with torch.no_grad():
         logits = model(**inputs).logits
     probs = torch.softmax(logits, dim=1)[0].tolist()
-
     return {
-        "label_scores": {
-            model.config.id2label[0]: round(probs[0], 4),
-            model.config.id2label[1]: round(probs[1], 4)
-        },
         "ai_generated_score": probs[1],
-        "human_written_score": probs[0],
-        "is_ai": probs[1] > probs[0]
+        "human_written_score": probs[0]
     }
 
 # =====================================================
@@ -70,24 +64,32 @@ async def analyze(data: InputText):
     if not text:
         return {"success": False, "code": 400, "message": "Empty input text"}
 
-    paragraphs = split_into_paragraphs(text)
+    total_words = len(text.split())
+    full_text_result = analyze_text_block(text)
+    fake_percentage = round(full_text_result["ai_generated_score"] * 100, 2)
+
     results = []
+    ai_words = int(total_words * (fake_percentage / 100))
+
+    # ✅ If overall text looks AI-generated, analyze paragraphs for details
+    if fake_percentage > 50:
+        paragraphs = split_into_paragraphs(text)
+        ai_words, total_words = 0, 0
 
-    ai_words, total_words = 0, 0
+        for paragraph in paragraphs:
+            res = analyze_text_block(paragraph)
+            word_count = len(paragraph.split())
+            total_words += word_count
+            ai_words += word_count * res["ai_generated_score"]
 
-    for paragraph in paragraphs:
-        res = analyze_text_block(paragraph)
-        results.append({
-            "paragraph": paragraph,
-            "ai_generated_score": res["ai_generated_score"],
-            "human_written_score": res["human_written_score"]
-        })
+            results.append({
+                "paragraph": paragraph,
+                "ai_generated_score": res["ai_generated_score"],
+                "human_written_score": res["human_written_score"]
+            })
 
-        word_count = len(paragraph.split())
-        total_words += word_count
-        ai_words += word_count * res["ai_generated_score"]
+        fake_percentage = round((ai_words / total_words) * 100, 2) if total_words > 0 else 0
 
-    fake_percentage = round((ai_words / total_words) * 100, 2) if total_words > 0 else 0
     feedback = (
         "Most of Your Text is AI/GPT Generated"
         if fake_percentage > 50
@@ -105,7 +107,7 @@ async def analyze(data: InputText):
             "h": [r["paragraph"] for r in results],
             "hi": [],
             "textWords": total_words,
-            "aiWords": int(total_words * (fake_percentage / 100)),
+            "aiWords": ai_words,
             "fakePercentage": fake_percentage,
             "specialIndexes": [],
             "specialSentences": [],