Update app.py

app.py

@@ -75,11 +75,15 @@ def check_server_ready(port):
         return False
 
 @spaces.GPU(duration=170)
-def generate_image(image):
+def generate_image(prompt, image):
     prefix_filename = str(random.randint(0, 999999))
-    prompt = read_prompt_from_file('config.json')
-    prompt = json.dumps(prompt, ensure_ascii=False).replace('output_image', prefix_filename)
-    prompt = json.loads(prompt)
+    try:
+        prompt = prompt.replace('ComfyUI', prefix_filename)
+        prompt = json.loads(prompt)
+    except:
+        prompt = read_prompt_from_file('config.json')
+        prompt = json.dumps(prompt, ensure_ascii=False).replace('output_image', prefix_filename)
+        prompt = json.loads(prompt)
 
     image = Image.fromarray(image)
     image.save(INPUT_DIR + '/input.png', format='PNG')
@@ -132,7 +136,10 @@ def generate_image(image):
 if __name__ == "__main__":
     demo = gr.Interface(
         fn=generate_image,
-        inputs=[gr.Image(image_mode='RGBA', type="numpy")],
+        inputs=[
+        "text",
+        gr.Image(image_mode='RGBA', type="numpy")
+    ],
         outputs=[gr.Image(type="numpy", image_mode='RGBA')],
         title="Image Upscaler",
         description="BEST UPSCALER EVER!!!!!"