Update app.py

app.py

@@ -15,32 +15,43 @@ WATERMARK_TEXT = "SelamGPT"
 MAX_RETRIES = 3
 TIMEOUT = 60
 EXECUTOR = ThreadPoolExecutor(max_workers=2)
+GEEZ_FONT_PATH = "fonts/NotoSansEthiopic-Bold.ttf"
+
+# ===== AUTHENTIC ETHIOPIAN THEME =====
+theme = gr.themes.Default(
+    primary_hue="green",
+    secondary_hue="yellow",
+    neutral_hue="red",
+    font=[gr.themes.GoogleFont("Noto Sans Ethiopic"), "Arial", "sans-serif"]
+).set(
+    button_primary_background_fill="linear-gradient(90deg, #078C03 0%, #FCDD09 50%, #DA1212 100%)",  # Ethiopian flag gradient
+    button_primary_text_color="#ffffff",
+    button_secondary_background_fill="#FCDD09",  # Yellow
+    slider_color="#DA1212"  # Red
+)
 
 # ===== WATERMARK FUNCTION =====
 def add_watermark(image_bytes):
-    """Add watermark with optimized PNG output"""
     try:
         image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
         draw = ImageDraw.Draw(image)
         
-        font_size = 24
+        # Try Ge'ez font first, then default
         try:
-            font = ImageFont.truetype("Roboto-Bold.ttf", font_size)
+            font = ImageFont.truetype(GEEZ_FONT_PATH, 30)
         except:
-            font = ImageFont.load_default(font_size)
+            font = ImageFont.load_default(30)
         
+        # Calculate position
         text_width = draw.textlength(WATERMARK_TEXT, font=font)
-        x = image.width - text_width - 10
-        y = image.height - 34
+        x = image.width - text_width - 20
+        y = image.height - 40
         
-        draw.text((x+1, y+1), WATERMARK_TEXT, font=font, fill=(0, 0, 0, 128))
-        draw.text((x, y), WATERMARK_TEXT, font=font, fill=(255, 255, 255))
+        # Draw with Ethiopian flag colors (green outline, yellow fill)
+        draw.text((x+2, y+2), WATERMARK_TEXT, font=font, fill="#078C03")  # Green shadow
+        draw.text((x, y), WATERMARK_TEXT, font=font, fill="#FCDD09")  # Yellow text
         
-        # Convert to optimized PNG
-        img_byte_arr = io.BytesIO()
-        image.save(img_byte_arr, format='PNG', optimize=True, quality=85)
-        img_byte_arr.seek(0)
-        return Image.open(img_byte_arr)
+        return image
     except Exception as e:
         print(f"Watermark error: {str(e)}")
         return Image.open(io.BytesIO(image_bytes))
@@ -50,17 +61,29 @@ def generate_image(prompt):
     if not prompt.strip():
         return None, "⚠️ Please enter a prompt"
     
+    # Ethiopian cultural enhancement
+    cultural_enhancements = [
+        "Ethiopian style", "Habesha culture", "vibrant colors",
+        "traditional elements", "East African aesthetic"
+    ]
+    
+    enhanced_prompt = f"{prompt}, {', '.join(cultural_enhancements)}"
+    
+    params = {
+        "height": 1024,
+        "width": 1024,
+        "num_inference_steps": 30,
+        "guidance_scale": 8.0,
+        "negative_prompt": "western, non-African, cartoonish, low quality"
+    }
+    
     def api_call():
         return requests.post(
             API_URL,
             headers=headers,
             json={
-                "inputs": prompt,
-                "parameters": {
-                    "height": 1024,
-                    "width": 1024,
-                    "num_inference_steps": 30
-                },
+                "inputs": enhanced_prompt,
+                "parameters": params,
                 "options": {"wait_for_model": True}
             },
             timeout=TIMEOUT
@@ -87,37 +110,41 @@ def generate_image(prompt):
     
     return None, "⚠️ Failed after multiple attempts. Please try later."
 
-# ===== GRADIO THEME =====
-theme = gr.themes.Default(
-    primary_hue="emerald",
-    secondary_hue="amber",
-    font=[gr.themes.GoogleFont("Poppins"), "Arial", "sans-serif"]
-)
-
 # ===== GRADIO INTERFACE =====
-with gr.Blocks(theme=theme, title="SelamGPT Image Generator") as demo:
+with gr.Blocks(theme=theme, title="SelamGPT - Ethiopian Image Generator") as demo:
+    with gr.Row():
+        gr.Image("ethiopia_flag.png", height=80, show_label=False, show_download_button=False)  # Add your flag image
+    
     gr.Markdown("""
-    # 🎨 SelamGPT Image Generator
-    *Powered by Stable Diffusion XL (1024x1024 PNG output)*
+    <center>
+    <h1 style="color:#078C03">S</h1>
+    <h1 style="color:#FCDD09">E</h1>
+    <h1 style="color:#DA1212">L</h1>
+    <h1 style="color:#078C03">A</h1>
+    <h1 style="color:#FCDD09">M</h1>
+    <h1 style="color:#DA1212">G</h1>
+    <h1 style="color:#078C03">P</h1>
+    <h1 style="color:#FCDD09">T</h1>
+    </center>
+    <center><i>Ethiopian AI Image Generator</i></center>
     """)
     
     with gr.Row():
         with gr.Column(scale=3):
             prompt_input = gr.Textbox(
-                label="Describe your image",
-                placeholder="A futuristic Ethiopian city with flying cars...",
-                lines=3,
-                max_lines=5
+                label="Describe your image (English or Amharic)",
+                placeholder="ሰላም! Describe an Ethiopian scene...",
+                lines=3
             )
             with gr.Row():
-                generate_btn = gr.Button("Generate Image", variant="primary")
+                generate_btn = gr.Button("Generate", variant="primary")
                 clear_btn = gr.Button("Clear")
             
             gr.Examples(
                 examples=[
-                    ["An ancient Aksumite warrior in cyberpunk armor, 4k detailed"],
-                    ["Traditional Ethiopian coffee ceremony in zero gravity"],
-                    ["Portrait of a Habesha queen with golden jewelry"]
+                    ["አዲስ አበባ በሌሊት ፣ የኢትዮጵያ ባህላዊ መብራቶች", "Addis Ababa at night with Ethiopian lights"],
+                    ["Habesha cultural dress with intricate embroidery"],
+                    ["Simien Mountains landscape with gelada monkeys"]
                 ],
                 inputs=prompt_input
             )