Delete all.py with huggingface_hub

all.py

@@ -1,233 +0,0 @@
-import cv2
-from PIL import Image
-import numpy as np
-import argparse
-import io
-import base64
-import time
-import logging
-from android import AndroidEditor
-from iphone import IPhoneEditor
-from bottom import IPhoneBottom, AndroidBottom 
-# Setup logging
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
-
-class UnifiedEditor:
-    def __init__(self, iphone_font_path="SFUIText-Regular.otf", android_font_path="Roboto-Regular.ttf"):
-        self.iphone_editor = IPhoneEditor(iphone_font_path)
-        self.android_editor = AndroidEditor(android_font_path)
-        self.iphone_bottom = IPhoneBottom(iphone_font_path)
-        self.android_bottom = AndroidBottom(android_font_path)
-
-    @staticmethod
-    def parse_anggota(anggota_str, ocr_count=None):
-        """Parse anggota parameter and handle + prefix for addition or return string as is"""
-        if anggota_str.startswith('+'):
-            # If starts with +, add to existing OCR count
-            if ocr_count is None:
-                logger.error("OCR count is None, cannot perform addition")
-                return None
-            try:
-                addition = int(anggota_str[1:])
-                result = ocr_count + addition
-                logger.info(f"Adding {addition} to OCR count {ocr_count} = {result}")
-                return result
-            except ValueError:
-                logger.error(f"Invalid number format in anggota: {anggota_str}")
-                return None
-        else:
-            # Try to convert to number first
-            try:
-                result = int(anggota_str)
-                logger.info(f"Using direct anggota value: {result}")
-                return result
-            except ValueError:
-                # If not a number, treat as string
-                logger.info(f"Using string anggota value: {anggota_str}")
-                return anggota_str
-
-    def detect_platform(self, text_list):
-        # """Deteksi platform berdasarkan pola teks"""
-        for i, text in enumerate(text_list):
-            if text.lower() in {"anggota", "members"} and i > 0 and text_list[i-1].isdigit():
-                print(text_list[i-1])
-                if not any(text_list[j] in {"-", "Grup"} for j in range(i-2, max(-1, i-5), -1)):
-                    return "Bottom iPhone" if any(t.lower() == "edit" for t in text_list) else "Bottom Android"
-        
-        window_size = 5
-        for i in range(len(text_list) - window_size + 1):
-            window = " ".join(text_list[i:i+window_size]).lower()
-            if ("info" in window and "grup" in window) or ("info" in window and "group" in window):
-                return "iPhone"
-        return "Android"
-
-    def process_image(self, image_path, anggota):
-        start_time = time.time()
-        image = cv2.imread(image_path)
-        if image is None:
-            logger.error("Failed to read image")
-            return None, None, None
-        
-        # Convert image to bytes for platform detection and OCR
-        _, img_encoded = cv2.imencode('.png', image)
-        img_bytes = img_encoded.tobytes()
-        
-        # Perform OCR once and reuse results
-        extracted_data = self.iphone_editor._perform_ocr(img_bytes)
-        if extracted_data is None:
-            logger.error("OCR failed")
-            return None, None, None
-        
-        text_list = extracted_data['text']
-        
-        # Detect platform using OCR results
-        platform = self.detect_platform(text_list)
-        logger.info(f"Detected platform: {platform}")
-        
-        # Get OCR count for potential addition
-        if platform == "Bottom iPhone":
-            ocr_count = self.iphone_bottom.replace_anggota(image, 0, skip_ocr=True, extracted_data=extracted_data, get_ocr_count_only=True)
-        elif platform == "Bottom Android":
-            ocr_count = self.android_bottom.replace_anggota(image, 0, skip_ocr=True, extracted_data=extracted_data, get_ocr_count_only=True)
-        else:
-            ocr_count = self._get_ocr_count_from_text_list(text_list)
-        
-        # Parse anggota parameter
-        parsed_anggota = self.parse_anggota(anggota, ocr_count)
-        if parsed_anggota is None:
-            logger.error("Invalid anggota parameter")
-            return None, None, None
-        
-        # Process with appropriate editor
-        if platform == "iPhone":
-            result, theme = self.iphone_editor._process_core(image, str(parsed_anggota), show_preview=True, return_theme=True, skip_ocr=True, extracted_data=extracted_data)
-        elif platform == "Bottom iPhone":
-            result, theme = self.iphone_bottom.replace_anggota(image, str(parsed_anggota), show_preview=True, return_theme=True, skip_ocr=True, extracted_data=extracted_data)
-        elif platform == "Bottom Android":
-            result, theme = self.android_bottom.replace_anggota(image, str(parsed_anggota), show_preview=True, return_theme=True, skip_ocr=True, extracted_data=extracted_data)
-        else:
-            result, theme = self.android_editor._process_core(image, str(parsed_anggota), show_preview=True, return_theme=True, skip_ocr=True, extracted_data=extracted_data)
-        
-        if result is None:
-            return None, None, None
-        
-        end_time = time.time()
-        logger.info(f"Total processing time: {end_time - start_time:.2f} seconds")
-        return result, theme, platform
-
-    def _get_ocr_count_from_text_list(self, text_list):
-        """Get OCR count from text list for addition operations"""
-        try:
-            # Try to find group index first
-            group_idx = None
-            for i, text in enumerate(text_list):
-                if "Grup" in text or "Group" in text:
-                    group_idx = i
-                    break
-            
-            if group_idx is not None:
-                # Find the original member count from OCR
-                for i in range(group_idx, min(group_idx + 5, len(text_list))):
-                    if text_list[i].isdigit():
-                        try:
-                            ocr_count = int(text_list[i])
-                            logger.info(f"Found OCR count: {ocr_count}")
-                            return ocr_count
-                        except ValueError:
-                            continue
-            
-            logger.warning("No valid OCR count found, using 0 as default")
-            return 0
-            
-        except Exception as e:
-            logger.error(f"Error getting OCR count: {str(e)}")
-            return 0
-
-    def process_image_bytes(self, image_bytes, anggota):
-        start_time = time.time()
-        try:
-            # Convert image_bytes to numpy array first
-            image_stream = io.BytesIO(image_bytes)
-            pil_image = Image.open(image_stream).convert('RGB')
-            image = np.array(pil_image)
-            
-            # Perform OCR once and reuse results
-            extracted_data = self.iphone_editor._perform_ocr(image_bytes)
-            if extracted_data is None:
-                logger.error("OCR failed")
-                return None, None, None
-            
-            text_list = extracted_data['text']
-            
-            # Detect platform using OCR results
-            platform = self.detect_platform(text_list)
-            logger.info(f"Detected platform: {platform}")
-            
-            # Get OCR count for potential addition - use 'image' (numpy array) instead of 'image_bytes'
-            if platform == "Bottom iPhone":
-                ocr_count = self.iphone_bottom.replace_anggota(image, 0, skip_ocr=True, extracted_data=extracted_data, get_ocr_count_only=True)
-            elif platform == "Bottom Android":
-                ocr_count = self.android_bottom.replace_anggota(image, 0, skip_ocr=True, extracted_data=extracted_data, get_ocr_count_only=True)
-            else:
-                ocr_count = self._get_ocr_count_from_text_list(text_list)
-            
-            # Parse anggota parameter
-            parsed_anggota = self.parse_anggota(anggota, ocr_count)
-            if parsed_anggota is None:
-                logger.error("Invalid anggota parameter")
-                return None, None, None
-            
-            # Process with appropriate editor
-            if platform == "iPhone":
-                result, theme = self.iphone_editor.process_image_bytes(image_bytes, str(parsed_anggota), extracted_data)
-            elif platform == "Bottom iPhone":
-                result, theme = self.iphone_bottom.replace_anggota(image, str(parsed_anggota), show_preview=True, return_theme=True, skip_ocr=True, extracted_data=extracted_data)
-                # Convert numpy array to base64 string for bottom platforms
-                if result is not None:
-                    pil_result = Image.fromarray(result)
-                    output_io = io.BytesIO()
-                    pil_result.save(output_io, format='PNG')
-                    result = base64.b64encode(output_io.getvalue()).decode('utf-8')
-            elif platform == "Bottom Android":
-                result, theme = self.android_bottom.replace_anggota(image, str(parsed_anggota), show_preview=True, return_theme=True, skip_ocr=True, extracted_data=extracted_data)
-                # Convert numpy array to base64 string for bottom platforms
-                if result is not None:
-                    pil_result = Image.fromarray(result)
-                    output_io = io.BytesIO()
-                    pil_result.save(output_io, format='PNG')
-                    result = base64.b64encode(output_io.getvalue()).decode('utf-8')
-            else:
-                result, theme = self.android_editor.process_image_bytes(image_bytes, str(parsed_anggota), extracted_data)
-            
-            if result is not None:
-                end_time = time.time()
-                logger.info(f"Total processing time: {end_time - start_time:.2f} seconds")
-                return result, theme, platform
-            else:
-                logger.error("Result image is None")
-                return None, None, None
-        except Exception as e:
-            logger.error(f"Error in process_image_bytes: {str(e)}")
-            return None, None, None
-
-if __name__ == '__main__':
-    parser = argparse.ArgumentParser(description='Proses gambar grup unified')
-    parser.add_argument('image_path', help='Path ke file gambar')
-    parser.add_argument('anggota', help='Jumlah anggota (bisa menggunakan + untuk menambah ke jumlah yang ada, contoh: +5 untuk menambah 5)')
-    args = parser.parse_args()
-    editor = UnifiedEditor()
-    result, theme, platform = editor.process_image(args.image_path, args.anggota)
-    if result is not None:
-        print(f"Processing completed successfully!")
-        print(f"Detected platform: {platform}")
-        print(f"Detected theme: {theme}")
-    else:
-        print("Processing failed!")
-
-# Contoh penggunaan:
-# python all.py image.png 10        # Set jumlah anggota menjadi 10
-# python all.py image.png +5        # Tambah 5 ke jumlah anggota yang ada di OCR
-# python all.py image.png +10       # Tambah 10 ke jumlah anggota yang ada di OCR
-# python all.py image.png "Banyak"  # Gunakan string "Banyak" sebagai anggota
-# python all.py image.png "100+"    # Gunakan string "100+" sebagai anggota