editing placeholder check

app.py

@@ -162,7 +162,6 @@ def show_elements(text):
 def show_elements_json_input(json_input):
     # Parse the JSON input
     data = json.loads(json_input)
-    print(data)
     
     outputs = []
     
@@ -172,8 +171,15 @@ def show_elements_json_input(json_input):
             continue
         
         description = details['description']
-        block_type, content = description.split(': ', 1)
         
+        # Split the description, defaulting to 'Unknown' if no colon is present
+        parts = description.split(': ', 1)
+        if len(parts) == 2:
+            block_type, content = parts
+        else:
+            block_type = 'Unknown'
+            content = description
+
         if block_type == 'UI':
             # Create HTML for UI elements
             ui_html = f'<div class="ui-element">{content}</div>'
@@ -182,15 +188,22 @@ def show_elements_json_input(json_input):
             # Display story elements as Markdown
             outputs.append(gr.Markdown(f"**{content}**"))
         elif block_type == 'Media':
-            if content.lower() == 'audio':
+            content_lower = content.lower()
+            if 'audio' in content_lower:
                 # Placeholder for audio element
                 outputs.append(gr.Audio(label=f"Audio: {content}"))
-            elif content.lower() == 'video':
+            elif 'video' in content_lower:
                 # Placeholder for video element
                 outputs.append(gr.Video(label=f"Video: {content}"))
-            elif content.lower() == 'image':
+            elif 'image' in content_lower:
                 # Placeholder for image element
                 outputs.append(gr.Image(label=f"Image: {content}"))
+            else:
+                # Default to text for unknown media types
+                outputs.append(gr.Markdown(f"*Media: {content}*"))
+        else:
+            # Handle unknown types as plain text
+            outputs.append(gr.Markdown(content))
     
     return outputs