Update main.py

main.py

@@ -142,21 +142,23 @@ prompt = ChatPromptTemplate.from_messages([
 ])
 
 def get_collection_files(collection_id: str, user_id: str) -> str:
-    """Get list of files in the specified collection"""
+    """Get list of files in the specified collection using the external API"""
     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()
+        url = f"https://pvanand-documind-api-v2.hf.space/rag/get_collection_files"
+        params = {
+            "collection_id": collection_id,
+            "user_id": user_id
+        }
+        headers = {
+            'accept': 'application/json'
+        }
+        response = requests.post(url, params=params, headers=headers, data='')
         
-        # Join the file names into a string
-        return ", ".join(unique_files)
+        if response.status_code == 200:
+            return response.text
+        else:
+            logging.error(f"Error from API: {response.text}")
+            return f"Error getting files: {response.text}"
     except Exception as e:
         logging.error(f"Error getting collection files: {str(e)}")
         return f"Error getting files: {str(e)}"