updata gradio logic: trying to solve generation error of current lyrics

app.py

@@ -550,12 +550,10 @@ def respond(message, state):
         if any(marker in response.lower() for marker in ["[verse", "[chorus", "[bridge"]):
             state.lyrics = response
         
-        # Return conversation for display
+        # Return conversation for display in messages format
         conversation_display = []
         for msg in state.conversation:
-            role = "User" if msg["role"] == "user" else "Assistant"
-            conversation_display.append([msg["content"] if msg["role"] == "user" else None, 
-                                       msg["content"] if msg["role"] == "assistant" else None])
+            conversation_display.append({"role": msg["role"], "content": msg["content"]})
         
         return "", conversation_display, state
         
@@ -563,11 +561,10 @@ def respond(message, state):
         error_response = f"Sorry, I encountered an error: {str(e)}"
         state.conversation.append({"role": "assistant", "content": error_response})
         
-        # Format conversation for display
+        # Format conversation for display in messages format
         conversation_display = []
         for msg in state.conversation:
-            conversation_display.append([msg["content"] if msg["role"] == "user" else None, 
-                                       msg["content"] if msg["role"] == "assistant" else None])
+            conversation_display.append({"role": msg["role"], "content": msg["content"]})
         
         return "", conversation_display, state
 
@@ -633,7 +630,7 @@ def build_interface():
                 # Chat interface
                 with gr.Group():
                     gr.Markdown("### 💬 Lyrics Creation Chat")
-                    chatbot = gr.Chatbot(height=400, label="AI Lyrics Assistant", show_copy_button=True)
+                    chatbot = gr.Chatbot(height=400, label="AI Lyrics Assistant", show_copy_button=True, type="messages")
                     
                     with gr.Row():
                         text_input = gr.Textbox(
@@ -868,11 +865,10 @@ def response_audio(state: AppState, audio: tuple, genre_value, mood_value, theme
 
         os.remove(file_name)
 
-    # Format conversation for display
+    # Format conversation for display in messages format
     conversation_display = []
     for msg in state.conversation:
-        conversation_display.append([msg["content"] if msg["role"] == "user" else None, 
-                                   msg["content"] if msg["role"] == "assistant" else None])
+        conversation_display.append({"role": msg["role"], "content": msg["content"]})
     
     return state, conversation_display