Update app.py

app.py

@@ -193,7 +193,7 @@ def view_selected_conversation(selection):
 # Function to delete selected conversation
 def delete_selected_conversation(selection):
     if not selection or selection == "Select a conversation":
-        return "Please select a conversation to delete.", gr.update()
+        return "Please select a conversation to delete.", gr.update(), ""
     
     try:
         # Extract conversation number from selection
@@ -215,13 +215,17 @@ def delete_selected_conversation(selection):
                 
                 # Update dropdown choices
                 choices = get_conversation_choices()
-                return f"Conversation {conv_number + 1} deleted successfully!", gr.update(choices=choices, value="Select a conversation")
+                
+                # Get updated all conversations display
+                all_convs = load_conversations()
+                
+                return f"Conversation {conv_number + 1} deleted successfully!", gr.update(choices=choices, value="Select a conversation"), all_convs
             else:
-                return "Conversation not found.", gr.update()
+                return "Conversation not found.", gr.update(), ""
         
-        return "No conversations file found.", gr.update()
+        return "No conversations file found.", gr.update(), ""
     except Exception as e:
-        return f"Error deleting conversation: {str(e)}", gr.update()
+        return f"Error deleting conversation: {str(e)}", gr.update(), ""
 
 # Function to clear current conversation
 def clear_conversation():
@@ -234,10 +238,10 @@ def delete_all_conversations():
     try:
         if os.path.exists("conversations.json"):
             os.remove("conversations.json")
-            return "All conversations deleted successfully!", gr.update(choices=[], value="Select a conversation")
-        return "No conversations to delete.", gr.update()
+            return "All conversations deleted successfully!", gr.update(choices=[], value="Select a conversation"), "No previous conversations found."
+        return "No conversations to delete.", gr.update(), "No previous conversations found."
     except Exception as e:
-        return f"Error deleting conversations: {str(e)}", gr.update()
+        return f"Error deleting conversations: {str(e)}", gr.update(), ""
 
 # Function to refresh dropdown
 def refresh_dropdown():
@@ -276,7 +280,8 @@ with gr.Blocks(title="Chat with Documents 💬 📚") as demo:
         with gr.Column(scale=3):
             chatbot = gr.Chatbot(
                 label="Chat",
-                height=400
+                height=400,
+                type="messages"
             )
             msg = gr.Textbox(
                 label="Your Question",
@@ -374,12 +379,12 @@ with gr.Blocks(title="Chat with Documents 💬 📚") as demo:
     delete_selected_btn.click(
         fn=delete_selected_conversation,
         inputs=conversation_dropdown,
-        outputs=[delete_status, conversation_dropdown]
+        outputs=[delete_status, conversation_dropdown, convs_display]
     )
     
     delete_all_btn.click(
         fn=delete_all_conversations,
-        outputs=[delete_status, conversation_dropdown]
+        outputs=[delete_status, conversation_dropdown, convs_display]
     )
 
 if __name__ == "__main__":