Update app.py

app.py

@@ -28,13 +28,31 @@ def create_repo(g, repo_name):
 
 def push_to_github(local_path, repo, github_token):
     repo_url = f"https://{github_token}@github.com/{repo.full_name}.git"
-    repo = Repo(local_path)
-    if 'origin' in [remote.name for remote in repo.remotes]:
-        origin = repo.remote('origin')
+    local_repo = Repo(local_path)
+    
+    if 'origin' in [remote.name for remote in local_repo.remotes]:
+        origin = local_repo.remote('origin')
         origin.set_url(repo_url)
     else:
-        origin = repo.create_remote('origin', repo_url)
-    origin.push(refspec='{}:{}'.format('master', 'master'))
+        origin = local_repo.create_remote('origin', repo_url)
+    
+    # Determine the current branch name
+    if not local_repo.heads:
+        # If the repo is empty, create and checkout a new branch
+        local_repo.git.checkout('-b', 'main')
+        current_branch = 'main'
+    else:
+        current_branch = local_repo.active_branch.name
+    
+    # Stage all files
+    local_repo.git.add(A=True)
+    
+    # Commit changes if there are any
+    if local_repo.is_dirty():
+        local_repo.git.commit('-m', 'Initial commit')
+    
+    # Push to the current branch
+    origin.push(refspec=f'{current_branch}:{current_branch}')
 
 def main():
     st.title("GitHub Repository Cloner and Uploader")