Update app.py

app.py

@@ -82,4 +82,57 @@ def generate_image(prompt):
         return None, f"⚠️ Error: {str(e)[:200]}"
 
 # ===== GRADIO INTERFACE =====
-# ... (keep your existing interface code exactly as is) ...
+with gr.Blocks(theme=theme, title="SelamGPT Image Generator") as demo:
+    gr.Markdown("""
+    # 🎨 SelamGPT Image Generator
+    *Powered by HiDream-I1-Full (1024x1024 PNG output)*
+    """)
+    
+    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
+            )
+            with gr.Row():
+                generate_btn = gr.Button("Generate Image", 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"]
+                ],
+                inputs=prompt_input
+            )
+            
+        with gr.Column(scale=2):
+            output_image = gr.Image(
+                label="Generated Image",
+                type="pil",
+                format="png",
+                height=512
+            )
+            status_output = gr.Textbox(
+                label="Status",
+                interactive=False
+            )
+    
+    generate_btn.click(
+        fn=generate_image,
+        inputs=prompt_input,
+        outputs=[output_image, status_output],
+        queue=True
+    )
+    
+    clear_btn.click(
+        fn=lambda: [None, ""],
+        outputs=[output_image, status_output]
+    )
+
+if __name__ == "__main__":
+    demo.queue(max_size=2)
+    demo.launch(server_name="0.0.0.0", server_port=7860)