Update app.py

app.py

@@ -67,16 +67,40 @@ def gpt_inference(base_url, model, openai_key, prompt):
     img = decode_image(img_b64)
     return img
 
-iface = gr.Interface(
-    fn=gpt_inference, 
-    inputs=[gr.components.Textbox(), gr.components.Dropdown(choices=["gpt-3.5-turbo", "gpt-4"], label="Model"), gr.components.Textbox(), gr.components.Textbox()], 
-    outputs=gr.Image(),
-    title="SolidUI AI-generated visualization platform",
-    description="""
-    AI-generated visualization prototyping and editing platform, support 2D, 3D models, combined with LLM(Large Language Model) for quick editing.
-    GitHub: https://github.com/CloudOrc/SolidUI
-    """,
-    labels=["Base URL", "Model", "OpenAI Key","Prompt"]
 
-)
+
+css = """
+body {
+    background-color: #F2F2F2;
+}
+"""
+
+title = """
+<h2>SolidUI AI-generated visualization platform</h2>
+<p>AI-generated visualization prototyping and editing platform Support 2D, 3D models, combined with LLM(Large Language Model) for quick editing.</p>
+<a target="_blank" href="https://github.com/CloudOrc/SolidUI">
+    <img src="https://img.shields.io/badge/-github-blue?logo=github" />
+</a>
+<a target="_blank" href="https://huggingface.co/spaces/CloudOrc/SolidUI">
+    <img src="https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face%20Spaces-blue" />
+</a>
+"""
+
+def layout_fn():
+    with gr.Blocks(css=css) as demo:
+        with gr.Column(elem_id="col-container"):
+            gr.HTML(title)
+            with gr.Form() as form:
+                form.inputs = [
+                    gr.components.Textbox(default="", label="Base URL"),
+                    gr.components.Dropdown(choices=["gpt-3.5-turbo", "gpt-4"], default="gpt-3.5-turbo", label="Model"),
+                    gr.components.Textbox(default="", label="OpenAI Key"),
+                    gr.components.Textbox(default="", label="Prompt")
+                ]
+                form.outputs = [gr.Image(label="Output Image")]
+                form.submit_button = gr.Button('Run')
+    return demo
+
+
+iface = gr.Interface(fn=gpt_inference, layout=layout_fn)
 iface.launch()