Update app.py

app.py

@@ -46,16 +46,15 @@ class InferenceClient:
 class MyChatbot(gr.Chatbot):
     """Custom Chatbot class for enhanced functionality."""
     def __init__(self, **kwargs):
-        super().__init__(**kwargs)
-        self.issues = []  # Store fetched issues
-        self.current_issue = None  # Store the currently selected issue
+        super().__init__(type="messages", **kwargs)  # Set type to 'messages'
+        self.issues = []
+        self.current_issue = None
 
     def postprocess(self, y):
         """Post-processes the response to handle commands and display results."""
         if not y:
             return []
             
-        # Handle different types of input
         if isinstance(y, dict):
             assistant_response = y.get('assistant_response', '')
             command = y.get('command', '')
@@ -63,7 +62,12 @@ class MyChatbot(gr.Chatbot):
             assistant_response = str(y)
             command = ''
 
-        return [(None, assistant_response)]
+        return [
+            {
+                "role": "assistant",
+                "content": assistant_response
+            }
+        ]
 
 # Update the Blocks implementation
 with gr.Blocks() as demo: