Update app.py

app.py

@@ -4,6 +4,9 @@ from __future__ import annotations
 
 import gradio as gr
 import torch
+import re
+import openai
+from cairosvg import svg2png
 
 from app_canny import create_demo as create_demo_canny
 from app_depth import create_demo as create_demo_depth
@@ -27,13 +30,59 @@ if not torch.cuda.is_available():
 
 model = Model(base_model_id=DEFAULT_MODEL_ID, task_name="Canny")
 
+def gpt_control(apikey, prompt):
+
+    openai.api_key = apikey
+
+    # gpt
+    messages = [{"role": "system", "content": "You are an SVG expert with years of experience and multiple contributions to the SVG project. Based on the prompt and the description, please generate the corresponding SVG code."},
+     {"role": "user", "content": f"""Provide only the shell command without any explanations.
+The current objective is below. Reply with the SVG code only:
+OBJECTIVE: {prompt}
+YOUR SVG CODE:
+"""}]
+
+    completion = openai.ChatCompletion.create(
+      model = "gpt-4",
+      messages = messages
+    )
+
+    chat_response = completion.choices[0].message.content
+
+    code = re.findall('<.*>', chat_response)
+    code_new = '\n'.join(code)
+
+    svg_code = f"""
+    {code_new}
+    """
+    svg2png(bytestring=svg_code,write_to='output.png')
+
+    return 'output.png'
+
+
 with gr.Blocks(css="style.css") as demo:
-    gr.Markdown(DESCRIPTION)
+    gr.HTML("<center>"
+            "<h1>🌁🪄🌃 - ControlNet with GPT-4</h1>"
+            "</center>")
+
+    gr.Markdown("## <center>🌟 Born to Create: Controllable Text-to-Image Generation with GPT-4</center>")
+    
     gr.DuplicateButton(
         value="Duplicate Space for private use",
         elem_id="duplicate-button",
         visible=SHOW_DUPLICATE_BUTTON,
     )
+    
+    with gr.Tab("GPT-4 Control"):
+        with gr.Row():
+            with gr.Column():
+                inp1 = gr.Textbox(label="OpenAI API Key", type="password")
+                inp2 = gr.Textbox(label="Position Prompt (as simple as possible)")
+                btn1 = gr.Button("GPT-4 Control", variant="primary")
+            with gr.Column():
+                out1 = gr.Image(label="Output Image", type="pil")
+        btn1.click(gpt_control, [inp1, inp2], [out1])
+
 
     with gr.Tabs():
         with gr.TabItem("Canny"):