Update app.py

app.py

@@ -52,7 +52,7 @@ logging.basicConfig(
 # -------------------------------------------------------------------------
 # 🤖 Roboflow configuration
 # -------------------------------------------------------------------------
-ROBOFLOW_API_KEY = "JKuvVbqDgv4mBdVM0fUE"  # ✅ correct key
+ROBOFLOW_API_KEY = "JKuvVbqDgv4mBdVM0fUE"  # ✅ your key
 PROJECT_NAME = "model_verification_project"
 VERSION_NUMBER = 2
 
@@ -60,129 +60,65 @@ VERSION_NUMBER = 2
 os.environ["ROBOFLOW_API_KEY"] = ROBOFLOW_API_KEY
 
 # -------------------------------------------------------------------------
-# ✍️ Handwriting model (Hugging Face Space) configuration
+# ✍️ Handwriting model (Hugging Face Space)
 # -------------------------------------------------------------------------
 HANDWRITING_MODEL_ENDPOINT = "3morrrrr/Handwriting_Model_Inf"
 
-
-# -------------------------------------------------------------------------
-# Text and sizing configuration
-# -------------------------------------------------------------------------
-MIN_WIDTH_PERCENTAGE = 0.8   # minimum scaling for text width
-TEXT_SCALE_FACTOR = 1.2      # base scale factor for text resizing
-
 # -------------------------------------------------------------------------
-# Debugging setup
+# ⚙️ General configuration
 # -------------------------------------------------------------------------
+MIN_WIDTH_PERCENTAGE = 0.8
+TEXT_SCALE_FACTOR = 1.2
 DEBUG = True
 DEBUG_DIR = os.path.join(tempfile.gettempdir(), "debug_images")
 os.makedirs(DEBUG_DIR, exist_ok=True)
 
-logging.info(f"Debug images will be stored in: {DEBUG_DIR}")
+logging.info(f"Debug images stored in: {DEBUG_DIR}")
 logging.info(f"Using Roboflow project '{PROJECT_NAME}' (v{VERSION_NUMBER}) with API key ending in {ROBOFLOW_API_KEY[-4:]}")
 logging.info(f"Using handwriting model endpoint: {HANDWRITING_MODEL_ENDPOINT}")
 
+# -------------------------------------------------------------------------
+# 🧩 Helper functions
+# -------------------------------------------------------------------------
+def format_text_for_paper(text, paper_width):
+    """Auto-wrap text to fit detected paper area."""
+    pixels_per_char = 13
+    est_chars = max(10, int((paper_width * 0.8) / pixels_per_char))
+    char_limit = min(60, est_chars)
+    words = text.split(" ")
+    lines, line = [], ""
+    for word in words:
+        if len(line + " " + word) <= char_limit:
+            line += (" " if line else "") + word
+        else:
+            lines.append(line)
+            line = word
+    if line:
+        lines.append(line)
+    return "\n".join(lines)
 
-# Function to format text to fit within paper boundaries
-def format_text_for_paper(text, paper_width, char_limit=None):
-    """
-    Format text to fit within paper boundaries
-    
-    Parameters:
-    - text: The input text to format
-    - paper_width: Width of the paper in pixels
-    - char_limit: Optional character limit per line (default: calculated from paper width)
-    
-    Returns:
-    - Formatted text with appropriate line breaks
-    """
-    # Calculate character limit based on paper width if not provided
-    # This is an approximation - characters have different widths
-    if char_limit is None:
-        # Rough estimate: ~12-15 pixels per character for most fonts at reasonable size
-        pixels_per_char = 13
-        estimated_chars = max(10, int((paper_width * 0.8) / pixels_per_char))
-        char_limit = min(60, estimated_chars)  # Never exceed the 60 char model limit
-    
-    logging.debug(f"Formatting text with {char_limit} chars per line for paper width {paper_width}px")
-    
-    # Split text by existing newlines first
-    paragraphs = text.split('\n')
-    formatted_lines = []
-    
-    for paragraph in paragraphs:
-        # Skip empty paragraphs but preserve the line break
-        if not paragraph.strip():
-            formatted_lines.append('')
-            continue
-            
-        words = paragraph.split(' ')
-        current_line = words[0]
-        
-        for word in words[1:]:
-            # If adding this word would exceed the limit, start a new line
-            if len(current_line) + 1 + len(word) <= char_limit:
-                current_line += ' ' + word
-            else:
-                formatted_lines.append(current_line)
-                current_line = word
-        
-        # Don't forget the last line
-        if current_line:
-            formatted_lines.append(current_line)
-    
-    # Join lines with newlines
-    formatted_text = '\n'.join(formatted_lines)
-    logging.debug(f"Formatted text from {len(text)} chars into {len(formatted_lines)} lines")
-    
-    return formatted_text
-
-# Function to save debug images - updated to handle numpy arrays
-def save_debug_image(image, filename, boxes=None, text=None):
+def save_debug_image(image, filename, text=None):
+    """Save debug images for visualization."""
     if not DEBUG:
         return
-    
-    # Convert numpy array to PIL Image if needed
     if isinstance(image, np.ndarray):
-        if len(image.shape) == 2:  # Grayscale
-            debug_img = Image.fromarray(image).convert('RGBA')
-        else:  # RGB or RGBA
-            debug_img = Image.fromarray(image).convert('RGBA')
-    else:
-        debug_img = image.copy()
-        if debug_img.mode != 'RGBA':
-            debug_img = debug_img.convert('RGBA')
-    
-    # Create a drawing context
-    draw = ImageDraw.Draw(debug_img)
-    
-    # Draw boxes if provided
-    if boxes:
-        for label, box in boxes:
-            # Draw rectangle
-            draw.rectangle(box, outline=(255, 0, 0, 255), width=3)
-            # Draw label
-            draw.text((box[0], box[1] - 20), label, fill=(255, 0, 0, 255))
-    
-    # Draw text if provided
+        image = Image.fromarray(image)
     if text:
-        # Try to find a font
-        try:
-            from PIL import ImageFont
-            font = ImageFont.truetype("Arial", 24)
-        except:
-            font = None
-        
-        # Draw text at the top of the image
-        draw.rectangle([(0, 0), (debug_img.width, 60)], fill=(0, 0, 0, 128))
-        draw.text((10, 10), text, fill=(255, 255, 255, 255), font=font)
-    
-    # Save the image
-    debug_path = os.path.join(DEBUG_DIR, filename)
-    debug_img.save(debug_path)
-    logging.debug(f"Saved debug image: {debug_path}")
-    return debug_path
+        draw = ImageDraw.Draw(image)
+        draw.rectangle([(0, 0), (image.width, 60)], fill=(0, 0, 0, 128))
+        draw.text((10, 10), text, fill=(255, 255, 255))
+    path = os.path.join(DEBUG_DIR, filename)
+    image.save(path)
+    logging.debug(f"Saved debug image: {path}")
+    return path
 
+# -------------------------------------------------------------------------
+# 🧠 Load Roboflow + HuggingFace models
+# -------------------------------------------------------------------------
+rf = Roboflow(api_key=ROBOFLOW_API_KEY)
+project = rf.workspace().project(PROJECT_NAME)
+model = project.version(VERSION_NUMBER).model
+client = Client(HANDWRITING_MODEL_ENDPOINT)
 # Improved function to detect paper angle
 def detect_paper_angle(image, bounding_box):
     """