Update app.py

app.py

@@ -4,6 +4,7 @@ from PIL import Image
 import cv2
 import re
 import gradio as gr
+import numpy as np  # Ensure numpy is imported
 
 # Initialize the OCR reader
 reader = easyocr.Reader(['en'])
@@ -22,12 +23,28 @@ dimension_pattern = r"(\d+'\s?\d+\")|(\d+\.?\d*\s?[x×]\s?\d+\.?\d*)"  # Pattern
 
 # Function to extract room data with counts and measurements
 def extract_room_data(results, room_patterns, dimension_pattern):
-    # ... (same as before) ...
+    room_data = {}
+
+    for result in results:
+        text = result[1].lower()  # Extract the text from the OCR result
+        for room_type, pattern in room_patterns.items():
+            if re.search(pattern, text):
+                # Check if room type is already in the dictionary
+                if room_type not in room_data:
+                    room_data[room_type] = {"count": 0, "measurements": []}
+                room_data[room_type]["count"] += 1
+
+                # Find dimensions in the text
+                dimensions = re.findall(dimension_pattern, text)
+                if dimensions:
+                    room_data[room_type]["measurements"].extend(dimensions)
+
+    return room_data
 
 # Function to process the uploaded image
 def process_image(image):
     # Convert the Gradio image to PIL Image
-     image = Image.fromarray(image)
+    image = Image.fromarray(image)
 
     # Perform OCR
     results = reader.readtext(np.array(image), detail=0)
@@ -40,7 +57,7 @@ def process_image(image):
     for room_type, data in room_data_with_counts.items():
         output_text += f"- {room_type.capitalize()}:\n"
         output_text += f"  Count: {data['count']}\n"
-        output_text += f"  Measurements: {data['measurements']}\n"
+        output_text += f"  Measurements: {', '.join(data['measurements'])}\n"
 
     return output_text
 
@@ -54,4 +71,4 @@ iface = gr.Interface(
 )
 
 # Launch the interface
-iface.launch(share=True)  # Set share=True to get a shareable link
+iface.launch(share=True)  # Set share=True to get a shareable link