Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 import os
 import requests
 import base64
+import numpy as np
+from PIL import Image
 
 # GitHub personal access token (replace this with your actual token)
 GITHUB_TOKEN = "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"  # Use your own token!
@@ -9,40 +11,20 @@ OWNER = "LAMENTIS1"  # GitHub username or organization
 REPO = "face_recog"  # GitHub repository name
 BRANCH = "master"  # Branch to commit to (could be 'main' or any other branch)
 
-# Ensure the uploads directory exists
+# Ensure the uploads directory exists locally
 UPLOAD_FOLDER = 'uploads'
 if not os.path.exists(UPLOAD_FOLDER):
     os.makedirs(UPLOAD_FOLDER)
 
-def check_folder_exists(person_name):
-    api_url = f"https://api.github.com/repos/{OWNER}/{REPO}/contents/labeled_images/{person_name.replace(' ', '%20')}"
-    response = requests.get(api_url, headers={"Authorization": f"Bearer {GITHUB_TOKEN}"})
-    
-    if response.status_code == 404:
-        create_dummy_file(person_name)
-        return f"Folder 'labeled_images/{person_name}' created successfully."
+# Create the folder on the GitHub repo for the person
+def create_folder(person_name):
+    folder_path = os.path.join(UPLOAD_FOLDER, person_name)
+    # If the folder doesn't exist locally, create it
+    if not os.path.exists(folder_path):
+        os.makedirs(folder_path)
+        return f"Folder '{folder_path}' created locally."
     else:
-        return f"Folder 'labeled_images/{person_name}' already exists."
-
-def create_dummy_file(person_name):
-    api_url = f"https://api.github.com/repos/{OWNER}/{REPO}/contents/labeled_images/{person_name.replace(' ', '%20')}/.dummy"
-    data = {
-        "message": f"Create folder for {person_name}",
-        "branch": BRANCH,
-        "content": base64.b64encode(b"").decode(),  # Empty content for the dummy file
-    }
-    
-    headers = {
-        "Authorization": f"Bearer {GITHUB_TOKEN}",
-        "Content-Type": "application/json",
-    }
-
-    response = requests.put(api_url, json=data, headers=headers)
-    
-    if response.status_code == 201:
-        print(f"Folder 'labeled_images/{person_name}' created successfully.")
-    else:
-        print(f"Failed to create folder: {response.status_code} - {response.text}")
+        return f"Folder '{folder_path}' already exists."
 
 def upload_image(image_path, person_name, file_name):
     api_url = f"https://api.github.com/repos/{OWNER}/{REPO}/contents/labeled_images/{person_name.replace(' ', '%20')}/{file_name}"
@@ -66,13 +48,18 @@ def upload_image(image_path, person_name, file_name):
     return response.status_code == 201
 
 def upload_files_to_github(person_name, image1, image2):
-    image1_path = os.path.join(UPLOAD_FOLDER, image1.name)
-    image2_path = os.path.join(UPLOAD_FOLDER, image2.name)
+    # Create folder locally for the user
+    create_folder(person_name)
+
+    # Save images to temporary files
+    image1_path = os.path.join(UPLOAD_FOLDER, person_name, "1.jpg")
+    image2_path = os.path.join(UPLOAD_FOLDER, person_name, "2.jpg")
 
-    image1.save(image1_path)
-    image2.save(image2_path)
+    # Convert numpy.ndarray to an image file and save
+    Image.fromarray(image1).save(image1_path)
+    Image.fromarray(image2).save(image2_path)
 
-    success_message = check_folder_exists(person_name)
+    success_message = f"Folder '{person_name}' is ready."
 
     success1 = upload_image(image1_path, person_name, "1.jpg")
     success2 = upload_image(image2_path, person_name, "2.jpg")
@@ -98,4 +85,4 @@ iface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch(share=True)  # This will create a public link