Update app.py

app.py

@@ -52,11 +52,21 @@ def create_chat_widget():
         return chatblock
 
 def json_to_markdown(json_data):
-    return "\n\n".join(
-        f"**{item['role'].capitalize()}:** \n{item['content']}" if item['type'] == 'message' else
-        f"```{item['format']}\n{item['content']}\n```" if item['type'] == 'code' else
-        f"```\n{item['content']}\n```" for item in json_data if item['role'] != 'user'
-    )
+    markdown_text = []
+    for item in json_data:
+        if item['role'] != 'user':  # Skip user entries
+            if item['type'] == 'message':
+                # Check if 'content' key exists before accessing it
+                content = item.get('content', "No content available")
+                markdown_text.append(f"**{item['role'].capitalize()}:** \n{content}")
+            elif item['type'] == 'code':
+                # Assume 'content' key exists, consider adding a similar check if necessary
+                markdown_text.append(f"```{item['format']}\n{item['content']}\n```")
+            elif item['type'] == 'console':
+                # Check for 'content' in console type
+                content = item.get('content', "No content available")
+                markdown_text.append(f"```\n{content}\n```")
+    return "\n\n".join(markdown_text)
 
 with gr.Blocks() as demo:
     with gr.Tab("HEXON Chatbot Assignment"):