Update app.py

app.py

@@ -44,6 +44,32 @@ def extract_property_info(prop):
 def sort_properties_by_order(properties):
     ordered_properties = sorted(properties.items(), key=lambda x: x[1].get('x-order', float('inf')))
     return ordered_properties
+    
+
+def parse_outputs(data):
+    values = []
+
+    if isinstance(data, dict):
+        # Handle case where data is an object
+        dict_values = []
+        for value in data.values():
+            extracted_values = parse_values(value)
+            # For dict, we append instead of extend to maintain list structure within objects
+            if isinstance(value, list):
+                dict_values += [extracted_values]
+            else:
+                dict_values += extracted_values
+        return dict_values
+    elif isinstance(data, list):
+        # Handle case where data is an array
+        list_values = []
+        for item in data:
+            # Here we extend to flatten the list since we're already in an array context
+            list_values += parse_values(item)
+        return list_values
+    else:
+        # Handle primitive data types directly
+        return [data]
 
 def create_gradio_app(api_spec, api_url):
     inputs = []
@@ -110,18 +136,18 @@ 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")
+            
+            outputs = parse_outputs(json_response)
 
-            output_images = []
-            for output_uri in json_response["output"]:
-                if output_uri.startswith("data:image"):
+            for output in outputs:
+                if output.startswith("data:image"):
                     # Process as image
                     base64_data = output_uri.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_uri.startswith("data:audio"):
-                    # Process as audio
+                elif output.startswith("data:audio"):
                     base64_data = output_uri.split(",", 1)[1]
                     audio_data = base64.b64decode(base64_data)
                     audio_stream = io.BytesIO(audio_data)
@@ -129,8 +155,9 @@ def create_gradio_app(api_spec, api_url):
                     filename = f"{uuid.uuid4()}.wav"  # Change format as needed
                     with open(filename, "wb") as audio_file:
                         audio_file.write(audio_stream.getbuffer())
-                    return gr.update(visible=False), gr.update(visible=True, value=filename), gr.update(visible=False), keys
-            
+                    return gr.update(visible=False), gr.update(visible=True, value=filename), gr.update(visible=False), keys  
+                else:
+                    return gr.update(visible=False), gr.update(visible=False), gr.update(visible=True, value=output)
         else:
             raise gr.Error("The submission failed!")
     return gr.Interface(fn=predict, inputs=inputs, outputs=outputs)