Update app.py

app.py

@@ -32,9 +32,18 @@ def excute_udiff(diffusion_model_id, concept, steps):
     print(f"my input is diffusion_model_id: {diffusion_model_id}, concept: {concept}, steps: {steps}")
     result = requests.post('http://{}:{}/udiff'.format(myip, myport), json={"diffusion_model_id": diffusion_model_id, "concept": concept, "steps": steps})
     print(f"result: {result}")
-    result = result.text[1:-1]
+    # result = result.text[1:-1]
+    res1 = ""
+    res2 = ""
+    if response.status_code == 200:
+        response_json = response.json()
+        print(response_json)
+        res1 = response_json['input_prompt']
+        res2 = response_json['output_prompt']
+    else:
+        print(f"Request failed with status code {response.status_code}")
     
-    return result
+    return res1, res2
 
 
 css = '''
@@ -90,7 +99,7 @@ with gr.Blocks(css=custom_css) as demo:
             img2 = gr.Image("images/cheetah.jpg",label="Image Gnerated by Prompt of UnlearnDiffAtk",width=260,show_share_button=False,show_download_button=False)
             
 
-    start_button.click(fn=excute_udiff, inputs=[drop_model, drop, shown_columns_step], outputs=[text_ouput], api_name="udiff")
+    start_button.click(fn=excute_udiff, inputs=[drop_model, drop, shown_columns_step], outputs=[text_input, text_ouput], api_name="udiff")
 
 
 demo.queue().launch(server_name='0.0.0.0')