Create app.py

app.py

@@ -0,0 +1,88 @@
+import gradio as gr
+import pytesseract
+from pytesseract import Output
+from PIL import Image
+
+def validate_chat_origin(image):
+    if image is None:
+        return "No image uploaded.", "Error"
+
+    
+    try:
+        data = pytesseract.image_to_data(image, output_type=Output.DICT)
+    except Exception as e:
+        return f"OCR Error: {e}", "Error"
+
+    width, height = image.size
+    midpoint = width / 2
+    
+    
+    valid_lines = []
+    
+    
+    ignore_words = [
+        "end-to-end", "encrypted", "business", "account", "block", 
+        "add", "contacts", "today", "joined", "march", "report", "spam"
+    ]
+
+    n_boxes = len(data['text'])
+    
+    
+    for i in range(n_boxes):
+        text = data['text'][i].strip()
+        
+        
+        if not text or int(data['conf'][i]) < 50:
+            continue
+            
+        
+        if any(ignored in text.lower() for ignored in ignore_words):
+            continue
+            
+        
+        valid_lines.append({
+            "text": text,
+            "top": data['top'][i],
+            "left": data['left'][i],
+            "width": data['width'][i]
+        })
+
+    
+    valid_lines.sort(key=lambda x: x['top'])
+
+    if not valid_lines:
+        return "No chat messages detected (only headers found).", "Unknown"
+
+    
+    first_message = valid_lines[0]
+    
+    
+    
+    is_left_aligned = first_message['left'] < midpoint
+    
+    debug_info = (
+        f"Detected First Text: '{first_message['text']}'\n"
+        f"Position X: {first_message['left']} (Image Width: {width})\n"
+        f"Alignment: {'Left (Stranger)' if is_left_aligned else 'Right (Me)'}"
+    )
+
+    if is_left_aligned:
+        return f"✅ VALID: First message is from stranger.\n\n{debug_info}", "VALID"
+    else:
+        return f"❌ INVALID: First message is from you.\n\n{debug_info}", "INVALID"
+
+# Gradio Interface
+iface = gr.Interface(
+    fn=validate_chat_origin,
+    inputs=gr.Image(type="pil", label="Upload Chat Screenshot"),
+    outputs=[
+        gr.Textbox(label="Analysis Result"),
+        gr.Label(label="Validation Status")
+    ],
+    title="WhatsApp First Message Validator",
+    description="Validates if the conversation was initiated by the stranger (Left Aligned = Valid). If the first text is Right Aligned, it is Invalid.",
+    theme="soft"
+)
+
+if __name__ == "__main__":
+    iface.launch()