Update document_rag_router.py

document_rag_router.py

@@ -372,6 +372,28 @@ async def delete_collection(collection_id: str, user_id: str):
             detail=f"Unexpected error: {str(e)}"
         )
 
+@router.post("/get_collection_files")
+def get_collection_files(collection_id: str, user_id: str) -> str:
+    """Get list of files in the specified collection"""
+    try:
+        # Get the full collection name
+        collection_name = f"{user_id}_{collection_id}"
+        
+        # Open the table and convert to pandas
+        table = db.open_table(collection_name)
+        df = table.to_pandas()
+        logging.info(f"fetched chunks {str(df.head())}")
+        
+        # Get unique file names
+        unique_files = df['file_name'].unique()
+        
+        # Join the file names into a string
+        return ", ".join(unique_files)
+    except Exception as e:
+        logging.error(f"Error getting collection files: {str(e)}")
+        return f"Error getting files: {str(e)}"
+
+
 @router.post("/query_collection_tool")
 async def query_collection_tool(input_data: QueryInput):
     try: