Create app.py

app.py

@@ -0,0 +1,101 @@
+import gradio as gr
+import os
+import requests
+import base64
+
+# GitHub personal access token (replace this with your actual token)
+GITHUB_TOKEN = "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"  # Use your own token!
+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
+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."
+    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}")
+
+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}"
+
+    with open(image_path, "rb") as image_file:
+        encoded_image = base64.b64encode(image_file.read()).decode()
+
+    data = {
+        "message": f"Add {file_name} for {person_name}",
+        "branch": BRANCH,
+        "content": encoded_image,
+    }
+
+    headers = {
+        "Authorization": f"Bearer {GITHUB_TOKEN}",
+        "Content-Type": "application/json",
+    }
+
+    response = requests.put(api_url, json=data, headers=headers)
+
+    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)
+
+    image1.save(image1_path)
+    image2.save(image2_path)
+
+    success_message = check_folder_exists(person_name)
+
+    success1 = upload_image(image1_path, person_name, "1.jpg")
+    success2 = upload_image(image2_path, person_name, "2.jpg")
+
+    os.remove(image1_path)
+    os.remove(image2_path)
+
+    if success1 and success2:
+        success_message += " Images uploaded successfully!"
+    else:
+        success_message += " Failed to upload images. Please try again."
+    
+    return success_message
+
+def gradio_interface(person_name, image1, image2):
+    return upload_files_to_github(person_name, image1, image2)
+
+# Create a Gradio interface
+iface = gr.Interface(
+    fn=gradio_interface,
+    inputs=[gr.Textbox(label="Person's Name"), gr.Image(label="Image 1"), gr.Image(label="Image 2")],
+    outputs="text"
+)
+
+if __name__ == "__main__":
+    iface.launch()