Update app.py

app.py

@@ -2,29 +2,43 @@ import gradio as gr
 from PIL import Image, ImageDraw, ImageFont
 
 def greet(name):
-    # Create an image with a blue background
-    img = Image.new('RGB', (500, 300), color=(100, 150, 255))
+    try:
+        # Create an image with a blue background
+        img = Image.new('RGB', (500, 300), color=(100, 150, 255))
+        
+        # Initialize the drawing context with the image object
+        draw = ImageDraw.Draw(img)
+        
+        # Path to the custom font file
+        font_path = "fonts/arial.ttf"  # Adjust the path as needed
+        
+        try:
+            # Try to use the custom font
+            font = ImageFont.truetype(font_path, 40)
+        except IOError:
+            # Fallback to default font if the custom font is not found
+            print("Custom font not found, using default font.")
+            font = ImageFont.load_default()
+        
+        # Define the text to be added to the image
+        text = f"Hello {name}!!"
+        
+        # Calculate the size of the text to center it
+        text_width, text_height = draw.textsize(text, font=font)
+        
+        # Calculate the x, y position of the text to be centered
+        x = (img.width - text_width) // 2
+        y = (img.height - text_height) // 2
+        
+        # Draw the text on the image
+        draw.text((x, y), text, fill=(255, 255, 255), font=font)  # White text color
+        
+        return img
     
-    # Initialize the drawing context with the image object
-    draw = ImageDraw.Draw(img)
-    
-    # Use the default font provided by PIL
-    font = ImageFont.load_default()
-    
-    # Define the text to be added to the image
-    text = f"Hello {name}!!"
-    
-    # Calculate the size of the text to center it
-    text_width, text_height = draw.textsize(text, font=font)
-    
-    # Calculate the x, y position of the text to be centered
-    x = (img.width - text_width) // 2
-    y = (img.height - text_height) // 2
-    
-    # Draw the text on the image
-    draw.text((x, y), text, fill=(255, 255, 255), font=font)  # White text color
-    
-    return img
+    except Exception as e:
+        # Print the error message for debugging
+        print(f"Error generating image: {e}")
+        return None  # Return None if there's an error to avoid breaking the app
 
 # Set up the Gradio interface
 demo = gr.Interface(fn=greet, inputs="text", outputs="image")