Update app.py

app.py

@@ -1,12 +1,23 @@
 import spaces
 import gradio as gr
 import torch
-from PIL import Image
+from PIL import Image, PngImagePlugin
 from diffusers import DiffusionPipeline
 import random
 import os
+import pygsheets
 from datetime import datetime
 
+# Initialize GSheet Connexion
+    #Authorization
+    gc = pygsheets.authorize(service_account_env_var='GSHEET_AUTH')
+
+    #Open the google spreadsheet
+    sh = gc.open('AndroFLUX-Logs')
+
+    #Select the first sheet 
+    wks = sh[0]
+
 # Initialize the base model and specific LoRA
 base_model = "black-forest-labs/FLUX.1-dev"
 pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
@@ -53,14 +64,24 @@ def run_lora(prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
     image_filename = f"generated_image_{timestamp}.png"
     image_path = os.path.join("/tmp/gradio", image_filename)
-    image.save(image_path)
+
+        # Add Metadata
+        new_metadata_string = f"{prompt}\nNegative prompt:  none \nSteps: {steps}, CFG scale: {cfg_scale}, Seed: {seed}, Lora hashes: AndroFlux-v19: c44afd41ece1"
+        metadata = PngImagePlugin.PngInfo()
+        metadata.add_text("parameters", new_metadata_string)
+
+    image.save(image_path, pnginfo=metadata)
 
     # Construct the URL to access the image
     space_url = "https://killwithabass-flux-1-dev-lora-androflux.hf.space"  # Replace with your actual space URL
     image_url = f"{space_url}/file={image_path}"
 
-    # Log the file URL
-    print(f"Image URL: {image_url}")
+    Log queries
+    try:
+        if "girl" not in prompt and "woman" not in prompt:
+            wks.append_table(values=[prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale,image_url])
+    except:
+        print(f"Image URL: {image_url}") # Log the file URL
 
     # Final update (100%)
     progress(100, "Completed!")