Update app.py

app.py

@@ -1,24 +1,45 @@
 from flask import Flask, request, jsonify, send_file
-import io
 import traceback
 from android import AndroidEditor
-import base64
+from iphone import IPhoneEditor
+import numpy as np
+import cv2
+import pytesseract
+import os
 
 app = Flask(__name__)
 
+ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif'}
+
+def allowed_file(filename):
+    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
+
 @app.route('/')
 def index():
+    """
+    Endpoint utama yang menampilkan pesan selamat datang
+    """
     return "Kelas Kink"
 
 @app.route('/android', methods=['POST'])
 def process_android_image():
+    """
+    Endpoint untuk memproses gambar dari Android
+    
+    Parameters:
+    - anggota: Nama anggota (form data)
+    - file: File gambar (form data)
+    
+    Returns:
+    - JSON response dengan status, theme, dan gambar yang diproses
+    """
     app.logger.info('Menerima permintaan POST untuk memproses gambar')
     anggota = request.form.get('anggota') 
     image_file = request.files.get('file') 
     
     if not anggota or not image_file:
         return jsonify({'status': 'failed', 'message': 'Parameter anggota dan file diperlukan'}), 400
-
+        
     try:
         file_bytes = image_file.read()
         editor = AndroidEditor()
@@ -34,159 +55,35 @@ def process_android_image():
 
 @app.route('/iphone', methods=['POST'])
 def process_iphone_image():
-    try :
-        return jsonify({'status': 'failed', 'message': 'Fitur ini sedang dikembangkan'}), 500
-    except Exception as e:
-        error_traceback = traceback.format_exc()
-        print(f"Error occurred: {error_traceback}")
-        return jsonify({
-            'status': 'failed',
-            'message': str(e),
-            'traceback': error_traceback
-        }), 500
-    # app.logger.info('Menerima permintaan POST untuk memproses gambar iPhone')
-    # anggota = request.form.get('anggota')
-    # image_file = request.files.get('file')
+    """
+    Endpoint untuk memproses gambar dari iPhone
     
-    # if not anggota or not image_file:
-    #     return jsonify({'status': 'failed', 'message': 'Parameter anggota dan file diperlukan'}), 400
-
-    # try:
-    #     # Read and process the uploaded image
-    #     file_bytes = image_file.read()
-    #     image_array = np.frombuffer(file_bytes, np.uint8)
-    #     image = cv2.imdecode(image_array, cv2.IMREAD_COLOR)
-        
-    #     if image is None:
-    #         raise ValueError("Gagal membaca gambar. File mungkin rusak atau format tidak didukung.")
-
-    #     extracted_data = pytesseract.image_to_data(image, output_type=pytesseract.Output.DICT)
-    #     text_list = extracted_data['text']
+    Parameters:
+    - anggota: Nama anggota (form data)
+    - file: File gambar (form data)
+    
+    Returns:
+    - JSON response dengan status, theme, dan gambar yang diproses
+    """
+    app.logger.info('Menerima permintaan POST untuk memproses gambar')
+    anggota = request.form.get('anggota') 
+    image_file = request.files.get('file') 
+    
+    if not anggota or not image_file:
+        return jsonify({'status': 'failed', 'message': 'Parameter anggota dan file diperlukan'}), 400
         
-    #     group_position = None
-    #     split_position = None
-    #     member_position = None
-    #     member_count_position = None
-    #     lang = ''
-
-    #     # Find the second occurrence of "Grup" or "Group"
-    #     group_indices = []
-    #     for i, text in enumerate(text_list):
-    #         if text in ["Grup", "Group"]:
-    #             group_indices.append(i)
-
-    #     if len(group_indices) >= 2:
-    #         i = group_indices[1]
-            
-    #         if (i + 3 < len(text_list) and (text_list[i+1] == "-") and text_list[i+3] in ["anggota", "member"]):
-    #             lang = 'id' if text_list[i] == "Grup" else 'en'
-                
-    #             group_position = {
-    #                 "left": extracted_data['left'][i],
-    #                 "top": extracted_data['top'][i],
-    #                 "width": extracted_data['width'][i],
-    #                 "height": extracted_data['height'][i],
-    #             }
-
-    #             split_position = {
-    #                 "left": extracted_data['left'][i + 1],
-    #                 "top": extracted_data['top'][i + 1],
-    #                 "width": extracted_data['width'][i + 1],
-    #                 "height": extracted_data['height'][i + 1],
-    #             }
-
-    #             member_position = {
-    #                 "left": extracted_data['left'][i + 3],
-    #                 "top": extracted_data['top'][i + 3],
-    #                 "width": extracted_data['width'][i + 3],
-    #                 "height": extracted_data['height'][i + 3],
-    #             }
-
-    #             if i + 2 < len(text_list):
-    #                 member_count_position = {
-    #                     "left": extracted_data['left'][i + 2],
-    #                     "top": extracted_data['top'][i + 2],
-    #                     "width": extracted_data['width'][i + 2],
-    #                     "height": extracted_data['height'][i + 2],
-    #                 }
-    #     else:
-    #         return jsonify({'status': 'failed', 'message': 'Pola grup tidak ditemukan pada gambar'}), 400
-
-    #     if member_position is None:
-    #         return jsonify({'status': 'failed', 'message': 'Pola anggota tidak ditemukan pada gambar'}), 400
-
-    #     # Get background color for masking
-    #     x = member_position['left'] + member_position['width'] + 100
-    #     y = member_position['top'] + member_position['height'] - 100
-    #     bg_color = image[y, x]
-    #     rgb = (int(bg_color[0]), int(bg_color[1]), int(bg_color[2]))
-
-    #     # Mask the original text areas
-    #     if group_position:
-    #         cv2.rectangle(
-    #             image,
-    #             (group_position['left'], group_position['top']),
-    #             (group_position['left'] + group_position['width'], group_position['top'] + group_position['height']),
-    #             rgb,
-    #             -1,
-    #         )
-
-    #     if split_position:
-    #         cv2.rectangle(
-    #             image,
-    #             (split_position['left'], split_position['top']),
-    #             (split_position['left'] + split_position['width'], split_position['top'] + split_position['height']),
-    #             rgb,
-    #             -1,
-    #         )
-
-    #     if member_position:
-    #         cv2.rectangle(
-    #             image,
-    #             (member_position['left'], member_position['top']),
-    #             (member_position['left'] + member_position['width'], member_position['top'] + member_position['height']),
-    #             rgb,
-    #             -1,
-    #         )
-
-    #     if member_count_position:
-    #         cv2.rectangle(
-    #             image,
-    #             (member_count_position['left'], member_count_position['top']),
-    #             (member_count_position['left'] + member_count_position['width'], member_count_position['top'] + member_count_position['height']),
-    #             rgb,
-    #             -1,
-    #         )
-
-    #         # Add new member count text
-    #         updated_member_count = {
-    #             'id': f"Grup · {anggota} anggota",
-    #             'en': f"Group · {anggota} member"
-    #         }.get(lang)
-
-    #         font_path = "SFNSText-Regular.otf"
-    #         font_size = max(member_count_position['width'], member_count_position['height']) + 5.5
-    #         font = ImageFont.truetype(font_path, int(font_size))
-
-    #         image_pil = Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
-    #         draw = ImageDraw.Draw(image_pil)
-    #         text_x = group_position['left']
-    #         text_y = member_count_position['top'] - 4
-    #         draw.text((text_x, text_y), updated_member_count, font=font, fill=(90, 94, 95, 255)) 
-    #         image = cv2.cvtColor(np.array(image_pil), cv2.COLOR_RGB2BGR)
-
-        # Encode and return the processed image
-        # _, img_encoded = cv2.imencode('.jpg', image)
-        # return jsonify(io.BytesIO(img_encoded.tobytes()), mimetype='image/jpeg')
-
-    # except Exception as e:
-    #     error_traceback = traceback.format_exc()
-    #     print(f"Error occurred: {error_traceback}")
-    #     return jsonify({
-    #         'status': 'failed',
-    #         'message': str(e),
-    #         'traceback': error_traceback
-    #     }), 500
+    try:
+        file_bytes = image_file.read()
+        editor = IPhoneEditor()
+        img_b64, theme = editor.process_image_bytes(file_bytes, anggota)
+        if img_b64 is None:
+            return jsonify({'status': 'failed', 'message': 'Gagal memproses gambar'}), 500
+        return jsonify({'status': 'success', 'theme': theme, 'image': img_b64})
+    except Exception as e:
+        error_message = str(e)
+        error_traceback = traceback.format_exc()
+        print("Error occurred:", error_traceback) 
+        return jsonify({'status': 'failed', 'message': error_message, 'traceback': error_traceback}), 500
 
 if __name__ == '__main__':
     app.run(port=7860, debug=True, host='0.0.0.0')