Update app.py

app.py

@@ -1184,7 +1184,15 @@ def handle_large_dataset(df, create_document,isDataFrame):
     
         #file_path = "output_data.xlsx"
         #download_url = repo.get_download_url(file_path)
-        create_file_HF(file_path)
+        global user_name
+        # Get today's date and current time
+        now = datetime.now()
+        
+        # Format the date and time as 'YYYY-MM-DD HH:MM'
+        formatted_date_time = now.strftime("%Y-%m-%d %H:%M")
+        print(formatted_date_time)
+        directory = user_name + "/" + formatted_date_time
+        create_file_HF(file_path, directory)
         print("File created in the home path")
        
         #save_file_to_hostinger('output_data.xlsx')
@@ -1197,7 +1205,7 @@ def handle_large_dataset(df, create_document,isDataFrame):
             #Logic to generate pdfs with employee name and account number
             for index, row in df.iterrows():
                 # Create a PDF for each row
-                create_pdf(row['COMPANY'], row['EMPLOYEE NAME'], row['ACCOUNT NUMBER'])
+                create_pdf(row['COMPANY'], row['EMPLOYEE NAME'], row['ACCOUNT NUMBER'],directory)
             create_document = 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."
         print(sample_table)
@@ -1211,20 +1219,14 @@ def handle_large_dataset(df, create_document,isDataFrame):
     else:
         return "Your query returns a large dataset which is not supported in the current version. Please try a different query."
 
-def create_file_HF(file_path):
+def create_file_HF(file_path,directory):
     from huggingface_hub import create_branch, create_tag
     from huggingface_hub import login
     from huggingface_hub import HfApi
     api = HfApi()
-    global user_name
+    
     repo_id = "Redmind/NewageNXTGPT_Repo_trial"
-    # Get today's date and current time
-    now = datetime.now()
     
-    # Format the date and time as 'YYYY-MM-DD HH:MM'
-    formatted_date_time = now.strftime("%Y-%m-%d %H:%M")
-    print(formatted_date_time)
-    directory = user_name + "/" + formatted_date_time
     
     
     
@@ -1243,9 +1245,10 @@ def create_file_HF(file_path):
     
     
     api.upload_file(path_or_fileobj=file_path, repo_id="Redmind/NewageNXTGPT_Repo_trial",revision = "test-branch", repo_type= "space", path_in_repo=directory)
+    reuturn directory
             
         
-def create_pdf(cname,ename,account_number):  
+def create_pdf(cname,ename,account_number, directory):  
 
     filled = FormWrapper("LOA_Sample_new.pdf").fill(
         {
@@ -1257,8 +1260,10 @@ def create_pdf(cname,ename,account_number):
         },
     )
     output_file_name =  f"documents\\{ename}_{cname}.pdf"
+    
     with open(output_file_name, "wb+") as output:
         output.write(filled.read())
+    create_file_HF(output_file_name, directory)
 
     repo_id = "Redmind/NewageNXTGPT"
     file_output=f"static/{output_file_name}"