Update app.py

app.py

@@ -4,11 +4,8 @@ import cv2
 import numpy as np
 from PIL import Image, ImageEnhance
 from ultralytics import YOLO
-from torchvision.transforms.functional import InterpolationMode
-from transformers import AutoModel, AutoTokenizer
-import torchvision.transforms as T
-from decord import VideoReader, cpu
-
+from backPrompt import main as main_b
+from frontPrompt import main as main_f
 
 model_path = "best.pt" 
 model = YOLO(model_path)
@@ -82,12 +79,13 @@ def crop_image(image, bounding_boxes):
 
 
 def vision_ai_api(image, doc_type):
-    """Dummy API call for Vision AI, returns a fake JSON response."""
-    return {
-        "document_type": doc_type,
-        "extracted_text": "Dummy OCR result for " + doc_type,
-        "confidence": 0.99
-    }
+
+    if doc_type == "front":
+        results = main_f(image)
+    if doc_type == "back":
+        results = main_b(image)
+        
+    return results
 
 
 def predict(image):