Update app.py

app.py

@@ -191,140 +191,147 @@ def process_android_image():
 
 
 @app.route('/iphone', methods=['POST'])
-def process_iphone_image(image_path, anggota):
+def process_iphone_image():
+    app.logger.info('Menerima permintaan POST untuk memproses gambar iPhone')
+    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:
-        try:
-            image = cv2.imread(image_path)
-            if image is None:
-                raise ValueError("Gagal membaca gambar. Pastikan path file benar.")
-        except Exception as e:
-            raise ValueError(f"Gagal membaca atau meng-decode gambar: {str(e)}") from e
-
-        try:
-            extracted_data = pytesseract.image_to_data(image, output_type=pytesseract.Output.DICT)
-        except Exception as e:
-            raise ValueError("Gagal melakukan ekstraksi teks dengan pytesseract.") from e
+        # 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']
+        
         group_position = None
         split_position = None
         member_position = None
         member_count_position = None
         lang = ''
 
-        try:
+        # 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],
+                }
 
-            group_indices = []
-            for i, text in enumerate(text_list):
-                if text in ["Grup", "Group"]:
-                    group_indices.append(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],
+                }
 
-            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],
-                    }
+                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],
+                }
 
-                    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],
+                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
 
-                    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],
-                    }
+        # 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]))
 
-                    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:
-                print("Pola grup tidak ditemukan atau tidak cukup")
-
-        except Exception as e:
-            raise ValueError(f"Gagal dalam memproses posisi teks: {str(e)}")
-
-        try:
-            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]))
-
-            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,
-                )
-
-                updated_member_count = {
-                    'id': f"Grup · {anggota} anggota",
-                    'en': f"Group · {anggota} member"
-                }.get(lang)
- 
+        # 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,
+            )
 
-                font_path = "SFNSText-Regular.otf"
-                font_size = max(member_count_position['width'], member_count_position['height']) + 5.5
-                font = ImageFont.truetype(font_path, font_size)
+        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,
+            )
 
-                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)
+            # 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))
 
-        except Exception as e:
-            raise ValueError(f"Gagal dalam memproses gambar: {str(e)}")
+            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 send_file(io.BytesIO(img_encoded.tobytes()), mimetype='image/jpeg')
+
     except Exception as e:
-        print(f"Error: {str(e)}")
+        error_traceback = traceback.format_exc()
+        print(f"Error occurred: {error_traceback}")
+        return jsonify({
+            'status': 'failed',
+            'message': str(e),
+            'traceback': error_traceback
+        }), 500
 
 if __name__ == '__main__':
-    app.run(port=5000, debug=True, host='0.0.0.0') 
+    app.run(port=7860, debug=True, host='0.0.0.0')