Update modules/audio_ldm.py

modules/audio_ldm.py

@@ -12,9 +12,8 @@ class audio_generation:
         num = random.randint(1, 1000)
         output_dict = {}
         output_dict["llm_output"] = f"Done. Audio has been generated and displayed. Stored as variable {num}"
-        output_dict["display"] = {"content": audio, "sampling_rate": 16000}
-        output_dict["metadata"] = prompt
-        output_dict["num_name"] = str(num)
+        output_dict["real_output"] = {"display": {"content": audio, "sampling_rate": 16000}, "metadata": prompt, "name": str(num)}
+        #output_dict["num_name"] = str(num)
         #output_dict["real_output"] = {"display": {"audio": audio, "sampling_rate": 16000}, "metadata": prompt}
         output_dict["type"] = "audio"
         return output_dict