Update app.py

app.py

@@ -1155,7 +1155,7 @@ def handle_large_dataset(df, create_document,isDataFrame):
     print(f"Total rows: {total_rows}")
     
     #docstatus = f"Download the complete dataset <a href='https://huggingface.co/spaces/Redmind/NewageNXTGPT/blob/main/assets/output.xlsx' download> here.</a>.There are total of {total_rows} rows."
-    docstatus = f"Download the complete dataset <a href='https://redmindtechnologies.com/RedMindGPT/output.xlsx' download> here.</a>.There are total of {total_rows} rows."
+    
     if total_rows < 4000:
 
        
@@ -1196,6 +1196,8 @@ def handle_large_dataset(df, create_document,isDataFrame):
         print(formatted_date_time)
         directory = user_name + "/" + formatted_date_time
         create_file_HF(file_path, directory,False)
+        dataset_link = get_download_link(directory,file_path)
+        docstatus = f"Download the complete dataset <a href={dataset_link} download> here.</a>.There are total of {total_rows} rows."
         print("File created in the home path")
        
         #save_file_to_hostinger('output_data.xlsx')
@@ -1217,7 +1219,9 @@ def handle_large_dataset(df, create_document,isDataFrame):
             print(zip_file_name)
             zip_files_in_folder(directory, zip_file_name)
             create_file_HF(zip_file_name,directory, False)
-            docstatus = f"Please download the complete dataset here: <a href='https://redmindtechnologies.com/RedMindGPT/output.zip' download>Download</a>. {total_rows} documents are created successfully."
+            link = get_download_link(directory,zip_file_name)
+            print(f"downloadable link: {link}")
+            docstatus = f"Please download the complete dataset here: <a href={link} download>Download</a>. {total_rows} documents are created successfully."
         print(sample_table)
         # 5. Return the summary and downloadable link
         #return f"""
@@ -1279,7 +1283,7 @@ def create_pdf(cname,ename,account_number, directory):
     
     return f"{output_file_name} is created successfully."
 
-def zip_files_in_folder(directory, output_zip):
+def zip_files_in_folder(directory_output, output_zip):
     # Initialize the ZIP file
     directory = "./"
     with zipfile.ZipFile(output_zip, 'w', zipfile.ZIP_DEFLATED) as zipf:
@@ -1290,7 +1294,7 @@ def zip_files_in_folder(directory, output_zip):
             if file.endswith(".pdf") and os.path.isfile(file_path):
                 # Add file to the ZIP archive
                 zipf.write(file_path, file)  # Save with its filename in the ZIP
-    print(f"ZIP file created: {output_zip}, {file_path}, {file}")
+    print(f"ZIP file created: {output_zip}, {output_zip}, {directory_output}")
 
 def directory_exists(repo_id, directory, token):
     try:
@@ -1306,11 +1310,11 @@ def directory_exists(repo_id, directory, token):
         print(f"Error checking directory existence: {e}")
         return False
 
-def get_download_link(file_path):
+def get_download_link(file_path,file_name):
     # Hugging Face Spaces uses this URL format for repository-hosted files
-    base_url = "https://huggingface.co/spaces/<your-space-id>/blob/main"
+    base_url = "https://huggingface.co/spaces/<your-space-id>/blob/test-branch"
     # Replace <your-space-id> with your actual Space name
-    return f"{base_url}/{file_path}"
+    return f"{base_url}/{file_path}/{file_name}"
 
 css = """