Update app_scribble.py

app_scribble.py

@@ -9,16 +9,19 @@ from utils import randomize_seed_fn
 
 def magic_prompt(prompt):
     openai.api_key = "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"
-
-    completion = openai.ChatCompletion.create(
-      model="gpt-3.5-turbo",
-      messages=[
-        {"role": "system", "content": "You are a rephraser."},
-        {"role": "user", "content": "You will be rephrasing the given details (details will be given in triple backticks) about an object/product. For example if “gray metal moka pot with black holder on wooden surface” is the given details, it should be rephrased as “moka pot, gray metal body, black holder, on wooden surface”. Output should be in coma separated form. Don’t add any comment and additional feature. Just give 1 output at a time. The given details “””{prompt}”””".format(prompt=prompt)}
-      ]
-    )
-
-    mag_prompt = str(completion.choices[0].message.content)
+    mag_prompt=""
+    if(prompt==""):
+        mag_prompt = "Please write a prompt first ✍️"
+    else:
+        completion = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=[
+            {"role": "system", "content": "You are a rephraser."},
+            {"role": "user", "content": "You will be rephrasing the given details (details will be given in triple backticks) about an object/product. For example if “gray metal moka pot with black holder on wooden surface” is the given details, it should be rephrased as “moka pot, gray metal body, black holder, on wooden surface”. Output should be in coma separated form. Don’t add any comment and additional feature. Just give 1 output at a time. The given details “””{prompt}”””".format(prompt=prompt)}
+          ]
+        )
+    
+        mag_prompt = str(completion.choices[0].message.content)
     return mag_prompt
 
 
@@ -80,7 +83,8 @@ def create_demo(process):
                     )
             with gr.Column():
                 result = gr.Gallery(label='Output',
-                                    show_label=False,
+                                    show_label=True,
+                                    preview=True,
                                     columns=2,
                                     object_fit='scale-down')
         inputs = [