Update app.py

app.py

@@ -37,10 +37,6 @@ body {
     max-width: 600px;
     margin: auto;
 }
-.gradio-container .wrap .text-gray-700 {
-    color: white !important;
-}
-
 button {
     background: linear-gradient(135deg, #ff416c, #ff4b4b);
     color: white;
@@ -66,14 +62,13 @@ input, textarea {
 }
 """
 
-demo = gr.Interface(
-    fn=generate_image,
-    inputs=gr.Textbox(label="Enter your prompt", value="A robot on a stallion."),
-    outputs=gr.Image(label="Generated Image by Agents Valley"),
-    title="🚀 Agents Valley AI Image Generator",
-    description="<span style='color: white;'>Enter a description and generate an AI image powered by Agents Valley!</span>",
-    theme=None,
-    css=css
-)
+with gr.Blocks(css=css) as demo:
+    gr.Markdown(
+        "<h2 style='color: white; font-size: 18px; font-weight: 500;'>Enter a description and generate an AI image powered by Agents Valley!</h2>"
+    )
+    prompt_input = gr.Textbox(label="Enter your prompt", value="A robot on a stallion.")
+    output_image = gr.Image(label="Generated Image by Agents Valley")
+    generate_button = gr.Button("Generate Image")
+    generate_button.click(fn=generate_image, inputs=prompt_input, outputs=output_image)
 
 demo.launch()