Save to Github

.env

@@ -0,0 +1 @@
+TOKEN = "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"

app.py

@@ -2,21 +2,57 @@ import gradio as gr
 from ultralytics import YOLO
 import os
 from datetime import datetime
+import requests
+from requests.auth import HTTPBasicAuth
+import base64
 import cv2
+from dotenv import load_dotenv
+import io
 
 model = YOLO('head_yolov8s.pt')
+load_dotenv()
+
+def upload_to_github(file_content, filepath):
+    username = "mohitsriv99"
+    repository = "hsamples"
+    branch = "main"
+    url = f"https://api.github.com/repos/{username}/{repository}/contents/{filepath}"
+    token = os.getenv("TOKEN")  # Using an environment variable for the token
+
+    encoded_file = base64.b64encode(file_content).decode("utf-8")
+
+    payload = {
+        "message": f"Upload image {filepath}",
+        "content": encoded_file,
+        "branch": branch
+    }
+
+    headers = {
+        "Authorization": f"token {token}"
+    }
+
+    response = requests.put(url, json=payload, headers=headers)
+    if response.status_code == 201:
+        print("File uploaded successfully to GitHub.")
+    else:
+        print(f"Failed to upload file. Status code: {response.status_code}, Response: {response.text}")
 
 def head_detect(path):
     if path is None:
         return None
     
     img = cv2.imread(path)
-    # Save the original image
-    if not os.path.exists('uploaded_images'):
-        os.makedirs('uploaded_images')
+
+    # Generate a timestamped filename
     current_time = datetime.now().strftime("%Y%m%d_%H%M%S")
-    filename = f'uploaded_images/{current_time}.jpg'
-    cv2.imwrite(filename, img)
+    filename = f"uploaded_images/{current_time}.jpg"
+
+    # Encode the image to a memory buffer instead of saving to disk
+    _, buffer = cv2.imencode('.jpg', img)
+    img_byte_array = io.BytesIO(buffer).getvalue()
+
+    # Upload the image to GitHub
+    upload_to_github(img_byte_array, filename)
 
     output = model(source=img)
     res = output[0].cpu().numpy()

requirements.txt

@@ -1,2 +1,3 @@
 ultralytics
-opencv-python
+opencv-python
+python-dotenv