Update app.py

app.py

@@ -95,7 +95,7 @@ def save_conversation():
     global conversation_history
     
     if not conversation_history:
-        return "No conversation to save."
+        return "No conversation to save.", gr.update()
     
     try:
         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
@@ -106,9 +106,31 @@ def save_conversation():
             }
             json.dump(conv_data, f)
             f.write("\n")
-        return "Conversation saved successfully!"
+        
+        # Update the dropdown choices
+        choices = get_conversation_choices()
+        return "Conversation saved successfully!", gr.update(choices=choices)
     except Exception as e:
-        return f"Error saving conversation: {str(e)}"
+        return f"Error saving conversation: {str(e)}", gr.update()
+
+# Function to get conversation choices for dropdown
+def get_conversation_choices():
+    if os.path.exists("conversations.json"):
+        try:
+            with open("conversations.json", "r") as f:
+                conversations = [json.loads(line) for line in f]
+            
+            choices = []
+            for i, conv in enumerate(conversations):
+                timestamp = conv.get("timestamp", "Unknown time")
+                # Format: "1. 2024-01-13 14:30:45"
+                formatted_time = datetime.strptime(timestamp, "%Y%m%d_%H%M%S").strftime("%Y-%m-%d %H:%M:%S")
+                choices.append(f"{i + 1}. {formatted_time}")
+            
+            return choices
+        except Exception as e:
+            return []
+    return []
 
 # Function to load previous conversations
 def load_conversations():
@@ -121,9 +143,10 @@ def load_conversations():
             for i, conv in enumerate(conversations):
                 conv_text += f"\n{'='*50}\nConversation {i + 1}\n{'='*50}\n"
                 timestamp = conv.get("timestamp", "Unknown time")
-                conv_text += f"Timestamp: {timestamp}\n\n"
+                formatted_time = datetime.strptime(timestamp, "%Y%m%d_%H%M%S").strftime("%Y-%m-%d %H:%M:%S")
+                conv_text += f"Timestamp: {formatted_time}\n\n"
                 
-                messages = conv.get("messages", conv)  # Handle old format
+                messages = conv.get("messages", conv)
                 for message in messages:
                     role = message.get('role', 'unknown')
                     content = message.get('content', '')
@@ -134,6 +157,72 @@ def load_conversations():
             return f"Error loading conversations: {str(e)}"
     return "No previous conversations found."
 
+# Function to view selected conversation
+def view_selected_conversation(selection):
+    if not selection or selection == "Select a conversation":
+        return "Please select a conversation to view."
+    
+    try:
+        # Extract conversation number from selection
+        conv_number = int(selection.split(".")[0]) - 1
+        
+        if os.path.exists("conversations.json"):
+            with open("conversations.json", "r") as f:
+                conversations = [json.loads(line) for line in f]
+            
+            if conv_number < len(conversations):
+                conv = conversations[conv_number]
+                conv_text = f"{'='*50}\n"
+                timestamp = conv.get("timestamp", "Unknown time")
+                formatted_time = datetime.strptime(timestamp, "%Y%m%d_%H%M%S").strftime("%Y-%m-%d %H:%M:%S")
+                conv_text += f"Conversation Timestamp: {formatted_time}\n"
+                conv_text += f"{'='*50}\n\n"
+                
+                messages = conv.get("messages", conv)
+                for message in messages:
+                    role = message.get('role', 'unknown')
+                    content = message.get('content', '')
+                    conv_text += f"{role.upper()}: {content}\n\n"
+                
+                return conv_text
+        
+        return "Conversation not found."
+    except Exception as e:
+        return f"Error viewing conversation: {str(e)}"
+
+# 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()
+    
+    try:
+        # Extract conversation number from selection
+        conv_number = int(selection.split(".")[0]) - 1
+        
+        if os.path.exists("conversations.json"):
+            with open("conversations.json", "r") as f:
+                conversations = [json.loads(line) for line in f]
+            
+            if conv_number < len(conversations):
+                # Remove the selected conversation
+                del conversations[conv_number]
+                
+                # Write back the remaining conversations
+                with open("conversations.json", "w") as f:
+                    for conv in conversations:
+                        json.dump(conv, f)
+                        f.write("\n")
+                
+                # Update dropdown choices
+                choices = get_conversation_choices()
+                return f"Conversation {conv_number + 1} deleted successfully!", gr.update(choices=choices, value="Select a conversation")
+            else:
+                return "Conversation not found.", gr.update()
+        
+        return "No conversations file found.", gr.update()
+    except Exception as e:
+        return f"Error deleting conversation: {str(e)}", gr.update()
+
 # Function to clear current conversation
 def clear_conversation():
     global conversation_history
@@ -145,10 +234,15 @@ def delete_all_conversations():
     try:
         if os.path.exists("conversations.json"):
             os.remove("conversations.json")
-            return "All conversations deleted successfully!"
-        return "No conversations to delete."
+            return "All conversations deleted successfully!", gr.update(choices=[], value="Select a conversation")
+        return "No conversations to delete.", gr.update()
     except Exception as e:
-        return f"Error deleting conversations: {str(e)}"
+        return f"Error deleting conversations: {str(e)}", gr.update()
+
+# Function to refresh dropdown
+def refresh_dropdown():
+    choices = get_conversation_choices()
+    return gr.update(choices=choices)
 
 # Create Gradio interface
 with gr.Blocks(title="Chat with Documents 💬 📚") as demo:
@@ -182,7 +276,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",
@@ -198,14 +293,38 @@ with gr.Blocks(title="Chat with Documents 💬 📚") as demo:
                 save_status = gr.Textbox(label="Save Status", interactive=False)
         
         with gr.Column(scale=1):
-            gr.Markdown("### Previous Conversations")
-            load_convs_btn = gr.Button("Load Previous Conversations")
+            gr.Markdown("### Conversation Management")
+            
+            with gr.Row():
+                load_convs_btn = gr.Button("Load All Conversations")
+                refresh_btn = gr.Button("🔄 Refresh List")
+            
             convs_display = gr.Textbox(
-                label="Conversation History",
-                lines=20,
+                label="All Conversations",
+                lines=10,
                 interactive=False
             )
-            delete_all_btn = gr.Button("Delete All Conversations", variant="stop")
+            
+            gr.Markdown("### Select & Manage Conversation")
+            
+            conversation_dropdown = gr.Dropdown(
+                label="Select Conversation",
+                choices=get_conversation_choices(),
+                value="Select a conversation",
+                interactive=True
+            )
+            
+            view_btn = gr.Button("View Selected", variant="secondary")
+            selected_conv_display = gr.Textbox(
+                label="Selected Conversation",
+                lines=10,
+                interactive=False
+            )
+            
+            with gr.Row():
+                delete_selected_btn = gr.Button("Delete Selected", variant="stop")
+                delete_all_btn = gr.Button("Delete All", variant="stop")
+            
             delete_status = gr.Textbox(label="Delete Status", interactive=False)
     
     # Event handlers
@@ -234,7 +353,7 @@ with gr.Blocks(title="Chat with Documents 💬 📚") as demo:
     
     save_btn.click(
         fn=save_conversation,
-        outputs=save_status
+        outputs=[save_status, conversation_dropdown]
     )
     
     load_convs_btn.click(
@@ -242,9 +361,26 @@ with gr.Blocks(title="Chat with Documents 💬 📚") as demo:
         outputs=convs_display
     )
     
+    refresh_btn.click(
+        fn=refresh_dropdown,
+        outputs=conversation_dropdown
+    )
+    
+    view_btn.click(
+        fn=view_selected_conversation,
+        inputs=conversation_dropdown,
+        outputs=selected_conv_display
+    )
+    
+    delete_selected_btn.click(
+        fn=delete_selected_conversation,
+        inputs=conversation_dropdown,
+        outputs=[delete_status, conversation_dropdown]
+    )
+    
     delete_all_btn.click(
         fn=delete_all_conversations,
-        outputs=delete_status
+        outputs=[delete_status, conversation_dropdown]
     )
 
 if __name__ == "__main__":