Update app.py

app.py

@@ -384,8 +384,20 @@ def get_logs():
         return f"Error reading logs: {str(e)}"
 
 def select_chunk(evt: gr.SelectData):
-    logger.info(f"Selected chunk: {evt.value}")
-    return evt.value[0], evt.value[1]
+    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
+            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"
 
 async def edit_cut(chunk_id, start, end):
     logger.info(f"edit_cut called with chunk_id: {chunk_id}, start: {start}, end: {end}")
@@ -394,10 +406,11 @@ async def edit_cut(chunk_id, start, end):
         uuid.UUID(chunk_id, version=4)
     except ValueError:
         logger.warning(f"Invalid chunk_id: {chunk_id} is not a valid UUID")
-        return "Error: Invalid chunk_id", ""
+        return "Error: Invalid chunk_id", "Invalid chunk_id selected"
     api = OpenAIApi(api_key=os.getenv("AZURE_OPENAI_API_KEY"))
     result = api.editor.cut(chunk_id, int(start), int(end))
-    return result, api.editor.diff(chunk_id, result)
+    diff = api.editor.diff(chunk_id, result) if "Error" not in result else ""
+    return result, diff
 
 async def edit_copy(chunk_id, start, end):
     logger.info(f"edit_copy called with chunk_id: {chunk_id}, start: {start}, end: {end}")