Update app.py

app.py

@@ -1,19 +1,13 @@
 import gradio as gr
 import random
 
-# Load the model
 model = gr.load("models/Purz/face-projection")
 
 def generate_image(text, seed):
     if seed is not None:
         random.seed(seed)
-
-    if text in [example[0] for example in examples]:
-        print(f"Using example: {text}")
-
     return model(text)
 
-# Examples for user guidance
 examples = [
     ["Humanoid Cat Warrior, Full View", None],
     ["Warhammer Sisterhood", None],
@@ -21,58 +15,64 @@ examples = [
     ["Fantasy dragon", None]
 ]
 
-# Custom CSS for gradient buttons and capsule shapes
-css = """
-/* General button styles */
-button {
-    border-radius: 25px; /* Capsule shape */
-    padding: 10px 20px;
-    font-size: 16px;
-    font-weight: bold;
-    color: white;
-    background: linear-gradient(135deg, #6D5BBA, #8A2387); /* Gradient color */
-    border: none;
-    cursor: pointer;
-    transition: background 0.3s ease;
+# Custom CSS for minimalist style, horizontal example layout, and styled prompt box
+custom_css = """
+/* Remove border and background color for the output image preview */
+.output-image {
+    border: none !important;
+    background: none !important;
 }
 
-button:hover {
-    background: linear-gradient(135deg, #8A2387, #6D5BBA); /* Gradient reverse on hover */
+/* Prompt textbox styling */
+input[type="text"] {
+    border: 2px solid #ddd;
+    border-radius: 8px;
+    padding: 12px;
+    font-size: 16px;
+    width: 100%;
+    background-color: #f9f9f9;
 }
 
-/* Slider handle style */
-input[type=range] {
-    accent-color: #8A2387; /* Gradient color accent */
+/* Examples arranged in a single line, without buttons */
+.gr-examples {
+    display: flex;
+    flex-direction: row;
+    justify-content: space-evenly;
+    gap: 10px;
+    margin-top: 10px;
 }
 
-/* Textbox styling */
-.gr-textbox {
-    border-radius: 8px;
-    border: 1px solid #DDD;
-    padding: 10px;
+.gr-examples span {
+    font-size: 14px;
+    color: #333;
+    cursor: pointer;
+    padding: 5px;
+    transition: all 0.2s ease;
 }
 
-/* Output image styling */
-.output_image {
-    border: 2px solid #6D5BBA;
-    border-radius: 10px;
-    padding: 10px;
-    background-color: #f9f9f9;
+/* Hover effect on examples */
+.gr-examples span:hover {
+    color: #007bff;
+    text-decoration: underline;
 }
 """
 
-# Gradio interface with custom styles
+# Function to handle example clicks (to simulate button-less selection)
+def update_prompt_text(example_text):
+    return example_text, None
+
+# Gradio Interface with updated UI elements and custom CSS
 interface = gr.Interface(
     fn=generate_image,
     inputs=[
-        gr.Textbox(label="Type your imagination:", placeholder="Type or select an example..."),
+        gr.Textbox(label="Imagine anything:", placeholder="Describe the scene...", interactive=True),
         gr.Slider(minimum=0, maximum=10000, step=1, label="Seed (optional)")
     ],
-    outputs=gr.Image(label="Generated Image"),
+    outputs=gr.Image(label="Generated Image", type="pil"),
     examples=examples,
-    css=css,  # Apply custom CSS styling
-    description="Note: The model is running on CPU; processing might take a bit longer. Thank you for your patience!"
+    css=custom_css,  # Apply custom CSS
+    description="Enter a creative prompt or choose an example to generate your AI-powered image. Model performance may vary due to CPU usage.",
+    live=True
 )
 
-# Launch the interface
 interface.launch()