Update app.py

app.py

@@ -6,34 +6,29 @@ from PIL import Image, ImageDraw, ImageFont
 text_pipe = pipeline("text-classification", model="karalif/myTestModel", return_all_scores=True)
 
 def draw_text_with_highlight(text):
-    # Assuming a comprehensive font file is in the same directory as the script
-    font_path = "path/to/your/font.ttf"  # Update this to the path of your font file
-    font_size = 24  # Adjust font size as needed
+    font_path = "path/to/your/font.ttf"  # Make sure this path is correct
+    font_size = 24
 
-    # Create an image with white background
     image = Image.new("RGB", (1000, 60), "white")
     draw = ImageDraw.Draw(image)
-    font = ImageFont.truetype(font_path, font_size)
 
-    # Split the text to find the third word
+    # Check if the font file exists and fallback to default if not
+    try:
+        font = ImageFont.truetype(font_path, font_size)
+    except IOError:
+        print(f"Font file not found: {font_path}. Falling back to default font.")
+        font = ImageFont.load_default()
+
     words = text.split()
-    
-    # Initialize starting position
     x, y = 10, 10
     if len(words) >= 3:
-        # Process each word to either draw normally or highlight
         for i, word in enumerate(words):
-            if i == 2:  # The third word is at index 2
-                # Draw background for highlight
+            if i == 2:  # Highlight the third word
                 width, height = draw.textsize(word, font=font)
                 draw.rectangle((x, y, x + width, y + height), fill='yellow')
-                draw.text((x, y), word, fill="black", font=font)
-            else:
-                draw.text((x, y), word, fill="black", font=font)
-            # Update x to draw next word. Adding a space width.
-            x += draw.textsize(word + " ", font=font)[0]
+            draw.text((x, y), word, fill="black", font=font)
+            x += draw.textsize(word + " ", font=font)[0]  # Update x position
     else:
-        # Draw the text as is if less than three words
         draw.text((10, 10), text, fill="black", font=font)
 
     return image