Update app.py

app.py

@@ -112,6 +112,10 @@ api= HfApi()
 token = os.getenv("HF_TOKEN")
 login(token=os.getenv("HF_TOKEN"))
 
+repo_id = os.getenv("repo_id")
+branch = os.getenv("branch")
+
+
 langfuse_handler = CallbackHandler()
 langfuse_handler.auth_check()  # Optional: Checks if the authentication is successful
 
@@ -1276,11 +1280,7 @@ def handle_large_dataset(df, create_document,isDataFrame):
         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,directory,document_created = False):
- 
-    
-    
-    repo_id = "Redmind/NewageNXTGPT_Repo_trial" 
-    
+    global repo_id, branch   
     
     if document_created:
         directory = directory + "/" + "document"
@@ -1292,13 +1292,9 @@ def create_file_HF(file_path,directory,document_created = False):
     else:
          directory = directory + "/" + file_path
    
+        
     
-    #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.")
-    
-    
-    api.upload_file(path_or_fileobj=file_path, repo_id="Redmind/NewageNXTGPT_Repo_trial",revision = "test-branch", repo_type= "space", path_in_repo=directory)
+    api.upload_file(path_or_fileobj=file_path, repo_id=repo_id,revision = branch, repo_type= "space", path_in_repo=directory)
     
             
         
@@ -1354,7 +1350,8 @@ def directory_exists(repo_id, directory, token):
 
 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/Redmind/NewageNXTGPT_Repo_trial/resolve/test-branch"
+    global repo_id,branch
+    base_url = "https://huggingface.co/spaces/" + repo_id + "/resolve/" + branch
     # Replace <your-space-id> with your actual Space name
     return f"{base_url}/{file_path}/{file_name}"