Update app.py

app.py

@@ -82,6 +82,7 @@ docstatus = ""
 sample_table = ""
 #This is to define the summary of the runtime tool. This summary will be updated in prompt template and description of the new tool
 run_time_tool_summary=""
+user_name = ""
 
 # Define global variables for managing the thread and current_event
 executor = concurrent.futures.ThreadPoolExecutor(max_workers=1)
@@ -845,6 +846,8 @@ def handle_dislike(data: gr.LikeData):
 
 # greet with user name on successful login
 def update_message(request: gr.Request):
+    global user_name
+    user_name = request.username
     return f"<h2 style=' font-family: Calibri;'>Welcome, {request.username}</h4>"
 
 # Function to generate a 50-word summary of the newly uploaded doc using OpenAI
@@ -1174,8 +1177,8 @@ def handle_large_dataset(df, create_document,isDataFrame):
         current_folder = os.getcwd()
         
 
-        
-        file_path = os.path.join(current_folder, 'output_data.xlsx')
+        file_path = "output_data.xlsx"
+        #file_path = os.path.join(current_folder, 'output_data.xlsx')
         df.to_excel(file_path, index=False)
               
     
@@ -1211,14 +1214,23 @@ def handle_large_dataset(df, create_document,isDataFrame):
 def create_file_HF(file_path):
     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"
+    directory = user_name
+    token = os.getenv("HF_TOKEN")
+    #check if direcotry exists
+    dir_exists_flag = directory_exists(repo_id, directory, token)
+    if not dir_exists_flag:
+        file_path = directory + "/" + file_path
+    #if directory does not exist, create the directoty
     login(token=os.getenv("HF_TOKEN"))
     #create_branch("Redmind/NewageNXTGPT_Repo_trial", repo_type="space", branch="test-branch")
     
     #create_tag("bigcode/the-stack", repo_type="dataset", revision="v0.1-release", tag="v0.1.1", tag_message="Bump release version.")
     
-    from huggingface_hub import HfApi
-    api = HfApi()
+    
     api.upload_file(path_or_fileobj=file_path, repo_id="Redmind/NewageNXTGPT_Repo_trial",revision = "test-branch", repo_type= "space", path_in_repo='output_data.xlsx')
             
         
@@ -1243,7 +1255,17 @@ def create_pdf(cname,ename,account_number):
     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."
-    
+
+
+def directory_exists(repo_id, directory, token):
+    try:
+        # List all files in the repository
+        files = api.list_repo_files(repo_id=repo_id, repo_type="space", token=token)
+        # Check if any file starts with the directory path
+        return any(file.startswith(directory) for file in files)
+    except Exception as e:
+        print(f"Error checking directory existence: {e}")
+        return False
 
 css = """