Upload folder using huggingface_hub

gradio_demo/app.py

@@ -32,22 +32,22 @@ from model_util import load_models_xl, get_torch_device, torch_gc
 import os
 import requests
 
-try:
-    # Send a GET request to the URL
-    response = requests.get("https://storage.googleapis.com/idfy-gff-public/idfy-gff-public%40idfy-eve-ml-training.iam.gserviceaccount.com.json")
+# try:
+#     # Send a GET request to the URL
+#     response = requests.get("https://storage.googleapis.com/idfy-gff-public/idfy-gff-public%40idfy-eve-ml-training.iam.gserviceaccount.com.json")
 
-    # Raise an exception if the request was unsuccessful
-    response.raise_for_status()
+#     # Raise an exception if the request was unsuccessful
+#     response.raise_for_status()
 
-    # Save the file to the specified path
-    with open("serviceaccount.json", 'wb') as file:
-        file.write(response.content)
+#     # Save the file to the specified path
+#     with open("serviceaccount.json", 'wb') as file:
+#         file.write(response.content)
 
-    print(f"Service account JSON file successfully downloaded")
-    os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = "serviceaccount.json"
+#     print(f"Service account JSON file successfully downloaded")
+os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = "serviceaccount.json"
 
-except requests.exceptions.RequestException as e:
-    print(f"Failed to download the service account JSON file: {e}")
+# except requests.exceptions.RequestException as e:
+#     print(f"Failed to download the service account JSON file: {e}")
 
 
 
@@ -273,46 +273,18 @@ def main(pretrained_model_name_or_path="wangqixun/YamerMIX_v8", enable_lcm_arg=F
     #     return p.replace("{prompt}", positive), n + ' ' + negative
     
     def store_images(email, gallery1, gallery2, gallery3, gallery4,consent):
+        if not email:
+            raise gr.Error("Email Id not provided")
         if not consent:
             raise gr.Error("Consent not provided")
-#         galleries = []
         for i, img in enumerate([gallery1, gallery2, gallery3, gallery4], start=1):
             if isinstance(img, np.ndarray):
                 img = Image.fromarray(img)
             print(f"Gallery {i} type after conversion: {type(img)}")
             dest = f'{email}/img{i}.png'
-            upload_pil_image_to_gcs(img,dest)
-#             galleries.append(img)
-#         # Create the images directory if it doesn't exist
-#         if not os.path.exists('images'):
-#             os.makedirs('images')
-
-#         # Define image file paths
-#         image_paths = []
-#         for i, img in enumerate(galleries, start=1):
-#             img_path = f'images/{email}_gallery{i}.png'
-#             img.save(img_path)
-#             image_paths.append(img_path)
-
-#         # Define the CSV file path
-#         csv_file_path = 'image_data.csv'
-
-#         # Create a DataFrame for the email and image paths
-#         df = pd.DataFrame({
-#             'email': [email],
-#             'img1_path': [image_paths[0]],
-#             'img2_path': [image_paths[1]],
-#             'img3_path': [image_paths[2]],
-#             'img4_path': [image_paths[3]],
-#         })
-
-#         # Write to CSV (append if the file exists, create a new one if it doesn't)
-#         if not os.path.isfile(csv_file_path):
-#             df.to_csv(csv_file_path, index=False)
-#         else:
-#             df.to_csv(csv_file_path, mode='a', header=False, index=False)
-        
+            upload_pil_image_to_gcs(img,dest) 
         gr.Info("Thankyou!! Your avatar is on the way to your inbox")
+        return None,None,None,None,None
     
     def add_watermark(image, watermark=logo, opacity=128, position="bottom_right", padding=10):
         # Convert NumPy array to PIL Image if needed
@@ -511,7 +483,7 @@ def main(pretrained_model_name_or_path="wangqixun/YamerMIX_v8", enable_lcm_arg=F
                     gallery3 = gr.Image(label="Generated Images", interactive=False)
                     gallery4 = gr.Image(label="Generated Images", interactive=False)
             email = gr.Textbox(label="Email", info="Enter your email address", value="")
-            consent = gr.Checkbox(label="I am giving my consent to use my data to share my AI Avtar and IDfy relevant information from time to time")
+            consent = gr.Checkbox(label="I am giving my consent to use my data to share my AI Avtar and IDfy relevant information from time to time", value=True)
             submit1 = gr.Button("SUBMIT", variant = "primary", elem_id="store-btn")
 #     with gr.Blocks(css=css) as demo:
 
@@ -589,7 +561,7 @@ def main(pretrained_model_name_or_path="wangqixun/YamerMIX_v8", enable_lcm_arg=F
             submit1.click(
                 fn=store_images,
                 inputs=[email,gallery1,gallery2,gallery3,gallery4,consent],
-                outputs=None)
+                outputs=[face_file,gallery1,gallery2,gallery3,gallery4])