Update Gradio app with multiple files

app.py

@@ -170,27 +170,27 @@ def chat_fn(message: Dict[str, Any], history: List[List[Any]]) -> Tuple[str, Lis
     
     return "", history
 
-def retry_fn(history: List[List[Any]]) -> Tuple[str, List[List[Any]]]:
+def retry_fn(history: List[Dict[str, Any]]) -> Tuple[str, List[Dict[str, Any]]]:
     """Retry the last message."""
-    if not history:
+    if not history or len(history) < 2:
         return "", history
     
     # Remove last assistant response and regenerate
-    last_user_msg = history[-1][0]
+    last_user_msg = history[-2]
     history = history[:-1]
     
     # Recreate the message dict
-    if isinstance(last_user_msg, dict):
-        message = {"text": last_user_msg.get("text", "")}
+    if isinstance(last_user_msg["content"], dict):
+        message = {"text": last_user_msg["content"].get("text", "")}
     else:
-        message = {"text": last_user_msg}
+        message = {"text": last_user_msg["content"]}
     
     return chat_fn(message, history)
 
-def undo_fn(history: List[List[Any]]) -> List[List[Any]]:
+def undo_fn(history: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
     """Undo the last message."""
     if history:
-        return history[:-1]
+        return history[:-2] if len(history) >= 2 else []
     return history
 
 def clear_fn() -> Tuple[None, List]:
@@ -245,7 +245,8 @@ with gr.Blocks(theme=gr.themes.Soft(), fill_height=True) as demo:
                 height=500,
                 show_copy_button=True,
                 bubble_full_width=False,
-                avatar_images=[None, "🤖"]
+                avatar_images=[None, "🤖"],
+                value=[]
             )
             
             with gr.Row():

requirements.txt

@@ -3,4 +3,4 @@ accelerate
 torch
 Pillow
 numpy
-torchvision
+torchvision