Update app.py

app.py

@@ -385,19 +385,23 @@ def get_logs():
 
 def select_chunk(evt: gr.SelectData):
     logger.info(f"Selected chunk raw data: {evt.value}")
-    # Ensure evt.value is a list and map to chunk_id and text based on headers order
-    if isinstance(evt.value, list) and len(evt.value) >= 2:
-        chunk_id = evt.value[0]  # First column is chunk_id
-        text = evt.value[1]      # Second column is text
-        try:
-            uuid.UUID(chunk_id, version=4)  # Validate chunk_id
+    # Handle single chunk_id or list of row data
+    chunk_id = evt.value if isinstance(evt.value, str) else (evt.value[0] if isinstance(evt.value, list) and len(evt.value) > 0 else "")
+    if not chunk_id:
+        logger.warning(f"Invalid selection data: No chunk_id found in {evt.value}")
+        return "", "Error: No chunk_id selected"
+    try:
+        uuid.UUID(chunk_id, version=4)  # Validate chunk_id
+        memory = ConversationMemory()
+        chunk = memory.get_chunk(chunk_id)
+        if chunk:
             logger.info(f"Selected chunk: {chunk_id}")
-            return chunk_id, text
-        except ValueError:
-            logger.warning(f"Invalid chunk_id selected: {chunk_id}")
-            return "", "Error: Invalid chunk_id selected"
-    logger.warning(f"Invalid selection data: {evt.value}")
-    return "", "Error: Invalid selection data"
+            return chunk_id, chunk["text"]
+        logger.warning(f"Chunk not found for chunk_id: {chunk_id}")
+        return "", "Error: Chunk not found"
+    except ValueError:
+        logger.warning(f"Invalid chunk_id selected: {chunk_id}")
+        return "", "Error: Invalid chunk_id selected"
 
 async def edit_cut(chunk_id, start, end):
     logger.info(f"edit_cut called with chunk_id: {chunk_id}, start: {start}, end: {end}")