Update app.py

app.py

@@ -123,50 +123,12 @@ css = """
     color: #4a5568;
     padding: 10px 16px;
 }
-
-.gr-accordion-panel {
-    padding: 0;
-}
-
-/* Footer Styles */
-.footer {
-    display: flex;
-    justify-content: center;
-    align-items: center;
-    margin-top: 3rem;
-    color: #4a5568;
-}
-
-.footer p {
-    margin-bottom: 0;
-    font-size: 0.875rem;
-}
-
 """
 
-with gr.Blocks(css=css) as demo:
+with gr.Interface(fn=generate_txt2img, inputs=[current_model, text_prompt, negative_prompt, image_style], outputs=image_output, css=css) as demo:
     favicon = '<img src="" width="48px" style="display: inline">'
-    gr.Markdown(
-        f"""<h1><center>{favicon} AI Diffusion</center></h1>
-            """
-    )
-
-    with gr.Row(elem_id="prompt-container"):
-        current_model = gr.Dropdown(label="Current Model", choices=list_models, value=list_models[1])
-
-    with gr.Row(elem_id="prompt-container"):
-        text_prompt = gr.Textbox(label="Prompt", placeholder="Enter a prompt", lines=1, elem_id="prompt-text-input")
-        text_button = gr.Button("Generate", variant='primary', elem_id="gen-button")
-
-    with gr.Row():
-        image_output = gr.Image(type="pil", label="Generated Image", elem_id="gallery")
-
-    with gr.Accordion("Advanced Settings", open=False):
-        negative_prompt = gr.Textbox(label="Negative Prompt", value="text, blurry, fuzziness", lines=1, elem_id="negative-prompt-text-input")
-        image_style = gr.Dropdown(label="Style", choices=["None style", "Cinematic", "Digital Art", "Portrait"], value="None style", allow_custom_value=False)
-
-    text_button.click(generate_txt2img, inputs=[current_model, text_prompt, negative_prompt, image_style], outputs=image_output)
-
-    gr.Footer("Powered by ngoctuanai", outbound_link="https://ngoctuanai-aidiffusion.hf.space")
+    demo.set_input_properties("current_model", label="Current Model")
+    demo.set_input_properties("text_prompt", label="Prompt", placeholder="Enter a prompt", lines=1)
+    demo.set_input_properties("text_button", label="Generate", button=True, primary=True)
 
-demo.launch()    
+    demo.launch(share=True, host='0.0.0.0', port=int(os.environ.get('PORT', 8080)))