Update app.py

app.py

@@ -12,11 +12,11 @@ def process(git_repo_url, space_destination, user_token):
     your_username = api.whoami(token=user_token)["name"]
 
     # Check if space ID is correct including username
-    if str(your_username) in str(space_destination):
+    if f"{your_username}/" in str(space_destination):
         print("username is present: good to continue")
     else:
         space_destination = f"{your_username}/{space_destination}"
-        print("user forgot to mention his username in space_destination")
+        print(f"user forgot to mention a username in space_destination: {space_destination}")
     
     # Create a temporary directory to clone the repository
     tmp_dir = tempfile.mkdtemp()
@@ -24,6 +24,15 @@ def process(git_repo_url, space_destination, user_token):
     # Clone the Hugging Face Spaces repository
     repo = Repo.clone_from(git_repo_url, tmp_dir)
 
+    # Check if README.md already exists in tmp_dir
+    readme_path = os.path.join(tmp_dir, "README.md")
+    original_readme_path = os.path.join(tmp_dir, "ORIGINAL_README.md")
+    
+    # Rename the existing README.md if it exists
+    if os.path.exists(readme_path):
+        print("Existing README.md found, renaming to ORIGINAL_README.md")
+        os.rename(readme_path, original_readme_path)
+
     # Log the contents of the tmp_dir with its subfolders
     print("Contents of the cloned repository:")
     for root, dirs, files in os.walk(tmp_dir):
@@ -41,13 +50,31 @@ def process(git_repo_url, space_destination, user_token):
             repo_type="space",
             token=user_token
         )
-        return "Done"
+        return f"Repository cloned ! Find it at hf.co/spaces/{space_destination}"
         
     except HfHubHTTPError as e:
         # Check if the error is a "Repository Not Found" error
         if "404 Client Error" in str(e) and "Repository Not Found" in str(e):
             print("Repository not found. Maybe we should Create the repository...")
-            return "Repository not found. Maybe we should Create the repository..."
+
+            api.create_repo(
+                repo_id=space_destination,
+                token=user_token,
+                private=True,
+                repo_type="space",
+                space_sdk="gradio"
+            )
+
+            print(f"New space repo created: {space_destination}")
+
+            api.upload_folder(
+                folder_path=tmp_dir,
+                repo_id=space_destination,
+                repo_type="space",
+                token=user_token
+            )
+                
+            return f"Repository cloned ! Find it at hf.co/spaces/{space_destination}"
         else:
             # Re-raise the exception if it's not a "Repository Not Found" error
             raise e