Update app.py

app.py

@@ -136,7 +136,8 @@ def create_gradio_app(api_spec, api_url):
             print(json_response)
             if "status" in json_response and json_response["status"] == "failed":
                 raise gr.Error("Failed to generate output")
-            
+
+            print(json_response["output"])
             outputs = parse_outputs(json_response["output"])
             print(outputs)
             for output in outputs:
@@ -144,13 +145,13 @@ def create_gradio_app(api_spec, api_url):
                     continue
                 if output.startswith("data:image"):
                     # Process as image
-                    base64_data = output_uri.split(",", 1)[1]
+                    base64_data = output.split(",", 1)[1]
                     image_data = base64.b64decode(base64_data)
                     image_stream = io.BytesIO(image_data)
                     image = Image.open(image_stream)
                     return gr.update(visible=True, value=image), gr.update(visible=False), gr.update(visible=False), keys
                 elif output.startswith("data:audio"):
-                    base64_data = output_uri.split(",", 1)[1]
+                    base64_data = output.split(",", 1)[1]
                     audio_data = base64.b64decode(base64_data)
                     audio_stream = io.BytesIO(audio_data)
                     # Here you can save the audio or return the stream for further processing