Update app.py

app.py

@@ -15,43 +15,32 @@ 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)
         
-        # Try Ge'ez font first, then default
+        font_size = 24
         try:
-            font = ImageFont.truetype(GEEZ_FONT_PATH, 30)
+            font = ImageFont.truetype("Roboto-Bold.ttf", font_size)
         except:
-            font = ImageFont.load_default(30)
+            font = ImageFont.load_default(font_size)
         
-        # Calculate position
         text_width = draw.textlength(WATERMARK_TEXT, font=font)
-        x = image.width - text_width - 20
-        y = image.height - 40
+        x = image.width - text_width - 10
+        y = image.height - 34
         
-        # 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
+        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))
         
-        return image
+        # 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)
     except Exception as e:
         print(f"Watermark error: {str(e)}")
         return Image.open(io.BytesIO(image_bytes))
@@ -61,29 +50,17 @@ 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": enhanced_prompt,
-                "parameters": params,
+                "inputs": prompt,
+                "parameters": {
+                    "height": 1024,
+                    "width": 1024,
+                    "num_inference_steps": 30
+                },
                 "options": {"wait_for_model": True}
             },
             timeout=TIMEOUT
@@ -110,41 +87,45 @@ def generate_image(prompt):
     
     return None, "⚠️ Failed after multiple attempts. Please try later."
 
-# ===== GRADIO INTERFACE =====
-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
-    
+# ===== GRADIO THEME ===== (add this right before the interface section)
+theme = gr.themes.Default(
+    primary_hue="green",
+    secondary_hue="yellow",
+    neutral_hue="red",
+    font=[gr.themes.GoogleFont("Poppins"), "Arial", "sans-serif"]
+).set(
+    button_primary_background_fill="#078C03",  # Ethiopian green
+    button_primary_text_color="#FCDD09",       # Ethiopian yellow
+    button_secondary_background_fill="#DA1212", # Ethiopian red
+    slider_color="#DA1212",                    # Red for sliders
+    checkbox_label_background_fill="#FCDD09",   # Yellow for checkboxes
+    block_background_fill_dark="#1e1e2e"       # Dark background
+)
+
+# ===== GRADIO INTERFACE ===== (keep your existing interface code)
+with gr.Blocks(theme=theme, title="SelamGPT Image Generator") as demo:
     gr.Markdown("""
-    <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>
+    # 🎨 SelamGPT Image Generator
+    *Powered by Stable Diffusion XL (1024x1024 PNG output)*
     """)
     
     with gr.Row():
         with gr.Column(scale=3):
             prompt_input = gr.Textbox(
-                label="Describe your image (English or Amharic)",
-                placeholder="ሰላም! Describe an Ethiopian scene...",
-                lines=3
+                label="Describe your image",
+                placeholder="A futuristic Ethiopian city with flying cars...",
+                lines=3,
+                max_lines=5
             )
             with gr.Row():
-                generate_btn = gr.Button("Generate", variant="primary")
+                generate_btn = gr.Button("Generate Image", variant="primary")
                 clear_btn = gr.Button("Clear")
             
             gr.Examples(
                 examples=[
-                    ["አዲስ አበባ በሌሊት ፣ የኢትዮጵያ ባህላዊ መብራቶች", "Addis Ababa at night with Ethiopian lights"],
-                    ["Habesha cultural dress with intricate embroidery"],
-                    ["Simien Mountains landscape with gelada monkeys"]
+                    ["An ancient Aksumite warrior in cyberpunk armor, 4k detailed"],
+                    ["Traditional Ethiopian coffee ceremony in zero gravity"],
+                    ["Portrait of a Habesha queen with golden jewelry"]
                 ],
                 inputs=prompt_input
             )