Update app.py

app.py

@@ -71,6 +71,10 @@ import base64
 #for PDF form filling
 from PyPDFForm import FormWrapper
 
+import os
+
+# Get the current working directory
+current_folder = os.getcwd()
 #Variables Initialization
 agent_executor = None
 vector_store1 = None
@@ -1152,12 +1156,8 @@ def handle_large_dataset(df, create_document,isDataFrame):
     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:
 
-        # 1. Limit to first 10 rows
        
-        
-        # 2. Handle missing values
-        #limited_data.fillna("N/A", inplace=True)
-        # 3. Drop the original first column
+        # 1. Drop the original first column
         if len(df.columns) > 1:
                 # Skipping the original first column
                 limited_data = df.head(3)
@@ -1165,8 +1165,8 @@ def handle_large_dataset(df, create_document,isDataFrame):
         else:
                 limited_data = df.head(20)
                 limited_data_without_first_column = limited_data 
-        #print( "range "+ len(limited_data_without_first_column))
-        # 4. Add SNo (serial number) as the first column, starting from 1
+
+        # 2. Add SNo (serial number) as the first column, starting from 1
         if isDataFrame :          
             
             limited_data_without_first_column.insert(0, 'SNo', range(1, len(limited_data_without_first_column) + 1))
@@ -1176,10 +1176,6 @@ def handle_large_dataset(df, create_document,isDataFrame):
         # 3. Save the full dataset to a downloadable file
         
        
-        import os
-
-        # Get the current working directory
-        current_folder = os.getcwd()
         
 
         file_path = "output_data.xlsx"
@@ -1187,8 +1183,8 @@ def handle_large_dataset(df, create_document,isDataFrame):
         df.to_excel(file_path, index=False)
               
     
-        #file_path = "output_data.xlsx"
-        #download_url = repo.get_download_url(file_path)
+       
+        
         global user_name
         # Get today's date and current time
         now = datetime.now()
@@ -1228,9 +1224,7 @@ def create_file_HF(file_path,directory):
  
     api = HfApi()
     
-    repo_id = "Redmind/NewageNXTGPT_Repo_trial"
-    
-    
+    repo_id = "Redmind/NewageNXTGPT_Repo_trial" 
     
     
     
@@ -1240,7 +1234,7 @@ def create_file_HF(file_path,directory):
         directory = directory + "/" + file_path
     else:
          directory = directory + "/" + file_path
-    #if directory does not exist, create the directoty
+   
     
     #create_branch("Redmind/NewageNXTGPT_Repo_trial", repo_type="space", branch="test-branch")
     
@@ -1268,11 +1262,7 @@ def create_pdf(cname,ename,account_number, directory):
         output.write(filled.read())
     create_file_HF(output_file_name, directory)
 
-    repo_id = "Redmind/NewageNXTGPT"
-    file_output=f"static/{output_file_name}"
-    from huggingface_hub import HfApi
-    api = HfApi()
-    #api.upload_file(path_or_fileobj=output_file_name, repo_id=repo_id, repo_type= "space", path_in_repo=file_output)
+    
     return f"{output_file_name} is created successfully."