Spaces:
Sleeping
Sleeping
File size: 2,524 Bytes
2f4af3f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 | from abc import ABC, abstractmethod
from PIL import Image
class BaseScriptProcessor(ABC):
def __init__(self, groq_client, references, clip_classifier=None): # Add clip_classifier parameter
self.groq_client = groq_client
self.references = references
self.clip_classifier = clip_classifier # Store clip_classifier
from services.rag_service import RAGService
from services.layout_parser import LayoutParser
self.rag_service = RAGService()
self.layout_parser = LayoutParser()
@abstractmethod
def detect_script(self, image_path):
"""Detect if image contains this script type"""
pass
@abstractmethod
def extract_text(self, image_path):
"""Extract text/symbols from image"""
pass
@abstractmethod
def process_text(self, extracted_text):
"""Process extracted text into meaningful output"""
pass
@abstractmethod
def generate_historical_context(self, processed_text):
"""Generate historical context for the text"""
pass
@abstractmethod
def generate_story(self, processed_text):
"""Generate creative story based on the text"""
pass
def process_image(self, image_path):
"""Main processing pipeline"""
try:
# Step 1: Detect script
is_detected, confidence = self.detect_script(image_path)
if not is_detected:
return None
# Step 2: Extract text
extracted_text = self.extract_text(image_path)
if not extracted_text:
return None
# Step 3: Process text
processed_result = self.process_text(extracted_text)
# Step 4: Generate context and story
historical_context = self.generate_historical_context(processed_result)
creative_story = self.generate_story(processed_result)
return {
"script_type": self.__class__.__name__.replace("Processor", "").lower(),
"confidence": confidence,
"extracted_text": extracted_text,
"processed_result": processed_result,
"historical_context": historical_context,
"creative_story": creative_story
}
except Exception as e:
print(f"[ERROR] Processing failed in {self.__class__.__name__}: {e}")
return None
|