Upload 9 files

.env

@@ -0,0 +1,8 @@
+REPLICATE_API_TOKEN="r8_Y2L8YCjCqtMnsoQDIfjZ9CL3Qllc9bH0M3nDY"
+BB_KeyID="0056550b44687420000000001"
+BB_AppKey= "K005EjPQyORkm9pYb97TKTqOLBFru3A"
+BB_bucket="Saiseisei"
+BB_defurl="https://f005.backblazeb2.com"
+DEEPL_Key="27b13bc8-a639-95b0-74f0-ea97eeda58a5:fx"
+email="saijapan2024@gmail.com"
+email_pw="awlgsztnxsscjlcf"

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+Tiles/4tile_20240115.png filter=lfs diff=lfs merge=lfs -text

Tiles/log.csv

@@ -0,0 +1,5 @@
+img,promt,ps
+./Tiles/4tile_20240115.png,Green garden with a water water fountain and a river,0.6
+./Tiles/Tile_20240123_1.jpg,Outer space dust cloud,0.6
+./Tiles/Tile_20240404.JPG,Volcano,0.6
+./Tile_20240421_1.jpg,Outer space dust cloud with a small ship flying through,0.6

app.py

@@ -0,0 +1,72 @@
+import gradio as gr
+from Saiseisei_workspace.utils import DeepL,Gen_image,send_email,base_model,example_data
+
+
+model_dic={}
+model_dic["None"]={
+    "model_loc":base_model,
+    "model_trigger":"high res, 4k",
+}
+
+model_dic["ISS MAP"]={
+    "model_loc":"jarvissan22/iss_nikond5:6ff0c8d11e33e4e8c91b1b9175a1c03f32fb559641bbbbe7007ac07b22fc5cec",
+    "model_trigger":"Aerial photo taken from the ISS photo taken from the ISS",
+}
+
+
+model_dic["Ukiyo-e_川瀬巴水"]={
+    "model_loc":"https://replicate.com/jarvissan22/kawasehasui_backgrounds",
+    "model_trigger":"Ukiyo-e,KawaseHasu",
+}
+
+model_dic["Anime_Galverse"]={
+    "model_loc":"galverse/setc-t1_label:65a7ee5a8c875fe9f38111699edf72f6c07f84dda7b7be5720e843ebb9f9c876",
+     "model_trigger":"anime coloring in the style of Galverse",
+}
+
+def gradio_gen_process(img,prompt,ps=0.6,style_model=None,ls=0.6,jp=None,recipient_email=None,model=base_model,model_trigger=None,save_loc=save_loc):
+  #JP Detect 
+  if jp:
+    prompt=DeepL(prompt)  
+  if style_model:
+    model=model_dic[style_model]["model_loc"]
+    model_trigger=model_dic[style_model]["model_trigger"]
+
+  if model_trigger:
+    prompt+=", "+model_trigger  
+  if type(ps)==str: # error on example detail: - input.prompt_strength: Invalid type. Expected: number, given: string
+    ps=float(ps)
+
+  #Image gen 
+  print(img)
+  img,file_saveloc=Gen_image(prompt,img=img,ps=ps,lr=ls,model=model,save_loc=save_loc) 
+  
+  #Send email
+  if recipient_email:
+    send_email(file_saveloc,recipient_email,prompt=prompt)
+    print("Email sent to ",recipient_email)
+  else:
+    print("Email not sent")
+
+  return img
+
+
+
+output_placeholder = gr.Label()
+gradio_interface=gr.Interface(
+    fn=gradio_gen_process,
+    inputs=[
+        gr.Image(value=example_data[0]["img"],type="filepath"),
+        gr.Textbox(label="Prompt"),
+        gr.Slider(minimum=0.1,maximum=1,value=0.6,label="PS:Image > Prompt"),
+        gr.Dropdown(list(model_dic.keys()),label="Style model"),
+        gr.Slider(minimum=0.1,maximum=1,value=0.6,label="LS:Style strength"),
+        gr.Checkbox(label="日本語", info="Promtは日本語ですか？"),
+        gr.Textbox(label="Email",placeholder="")
+    ],
+    outputs=[
+        gr.Image(label="gened image",type="pil")
+    ],
+    examples=img_loc
+)
+gradio_interface.launch(debug=True,share=True)

requirments.txt

@@ -0,0 +1,4 @@
+replicate==0.25.1
+gradio==4.26.0
+b2sdk==2.0.0
+dotenv-python

utils.py

@@ -0,0 +1,171 @@
+
+
+from IPython.display import display
+from PIL import Image
+import glob
+import gradio as gr
+import warnings
+import io
+import pandas as pd
+import replicate
+import json
+import requests
+import os
+from datetime import date
+import secrets
+import string
+import b2sdk.v2 as b2 #Backblaze img2img upload bucket 
+import os 
+import requests
+from dotenv import load_dotenv, find_dotenv
+_ = load_dotenv(find_dotenv())
+
+
+## Backclaze 
+info = b2.InMemoryAccountInfo()
+b2_api = b2.B2Api(info)
+application_key_id = os.getenv("BB_KeyID")
+application_key = os.getenv("BB_AppKey")
+#print(application_key_id,application_key)
+b2_api.authorize_account("production", application_key_id, application_key)     
+BB_bucket_name=os.getenv("BB_bucket")       
+BB_bucket=b2_api.get_bucket_by_name(os.getenv("BB_bucket"))
+BB_defurl=os.getenv("BB_defurl")
+
+
+def BB_uploadfile(local_file,file_name,b2_api=b2_api,BB_bucket_name=BB_bucket_name,FRIENDLY_URL=True):
+    metadata = {"key": "value"}
+    uploaded_file = BB_bucket.upload_local_file(
+            local_file=local_file,
+            file_name=file_name,
+            file_infos=metadata,
+    )
+    img_url=b2_api.get_download_url_for_fileid(uploaded_file.id_)
+    if FRIENDLY_URL: #Get friendly URP
+        img_url=BB_defurl+"/file/"+BB_bucket_name+"/"+file_name
+    print("backblaze", img_url)
+    return img_url
+
+
+#deepL api
+
+def DeepL(text,source_lang="JA",target_lang="EN",API_KEY= os.getenv("DEEPL_Key")):
+# パラメータの指定
+  params = {
+              'auth_key' : API_KEY,
+              'text' : text,
+              'source_lang' : source_lang, # 翻訳対象の言語
+              "target_lang": target_lang  # 翻訳後の言語
+          }
+  # リクエストを投げる
+  request = requests.post("https://api-free.deepl.com/v2/translate", data=params) # URIは有償版, 無償版で異なるため要注意
+  result = request.json()
+  return result["translations"][0]["text"]
+
+##Email client 
+
+import smtplib
+from email.mime.multipart import MIMEMultipart
+from email.mime.text import MIMEText
+from email.mime.image import MIMEImage
+from email.utils import formatdate
+
+
+
+def send_email(image_path, recipient_email,prompt=None,email=os.getenv("email"),pw=os.getenv("email_pw")):
+    sender_email = email
+    password = pw
+    msg = MIMEMultipart()
+    msg['From'] = sender_email
+    msg['To'] = recipient_email
+    msg['Subject'] =f"「Saiseisei Generated Image」: {os.path.basename(image_path)}"
+    msg['Date'] = formatdate()
+    # Attach prompt 
+    Text=f"""
+    再生成のワークショップをご参加ありがとうございました
+    生成したAI画像はこのメールに添付しました。
+    --
+    file name: {os.path.basename(image_path)}
+    prompt: {prompt}  
+    --
+    再生成について最も知りたいなら、IGで@pp_yokoと@jarvis_labsを連絡してください
+    https://www.instagram.com/jarvis_labs/
+    https://www.instagram.com/pp_yoko/
+    --
+    そして、SNSで画像を共有したいなら、ぜひしてお願いいたします
+    よろしくお願いいたします。
+    Jarvis
+    
+    """
+    msg.attach(MIMEText(Text, 'plain', 'utf-8'))
+    # Attach the image
+    with open(image_path, 'rb') as f:
+        img = MIMEImage(f.read())
+    msg.attach(img)
+    server = smtplib.SMTP('smtp.gmail.com', 587) #smtp_server = "smtp.gmail.com" Google
+    server.ehlo()
+    server.starttls()
+    server.ehlo()
+    server.login(sender_email, password)
+
+    server.send_message(msg)
+    server.quit()
+
+
+#Other
+def random_str(n=8):
+    # Define the characters to choose from
+    characters = string.ascii_letters + string.digits
+    # Generate the unreplicable string of length 8
+    unreplicable_string = ''.join(secrets.choice(characters) for _ in range(n))
+    return unreplicable_string
+
+
+#Replicate img gen
+base_model="lucataco/sdxl:c86579ac5193bf45422f1c8b92742135aa859b1850a8e4c531bff222fc75273d"
+
+img_loc="./Tiles/"  
+example_data=[]
+example_data.append({"img":img_loc+"4tile_20240115.png","promt":"Green garden with a water water fountain and a river","ps":0.6})
+example_data.append({"img":img_loc+"Tile_20240123_1.jpg","promt": "Outer space dust cloud","ps":0.6})
+example_data.append({"img":img_loc+"Tile_20240404.JPG","promt": "Volcano","ps":0.6})
+example_data.append({"img":img_loc+"Tile_20240421_1.jpg","promt": "Outer space dust cloud with a small ship flying through","ps":0.6})
+
+#"jarvissan22/kawasehasui_backgrounds:a4bb8bb1beb503b02c93789381544097dcb70afc92c512ea1500e70ccf704bc4"
+def Gen_image(prompt,img=None,ps=0.7,lr=0.8,model=base_model,save_loc="",w=1024,h=1024):
+  input={
+          "width": w,
+          "height": h,
+          "prompt": prompt,
+          "refine": "no_refiner",
+          "scheduler": "K_EULER",
+          "lora_scale": lr,
+          "num_outputs": 1,
+          "guidance_scale": 7.5,
+          "apply_watermark": True,
+          "high_noise_frac": 0.8,
+          "negative_prompt": "",
+          "prompt_strength": ps,
+          "num_inference_steps": 50
+      }
+
+  if img: #Upload image to backblaze 
+    img_url=BB_uploadfile(img,os.path.basename(img))
+    input["image"]=img_url
+
+  #Gen image
+  output = replicate.run(
+      model,
+      input=input,
+  )
+  #Get image
+  img_url=output[0]
+  r=requests.get(img_url)
+  image = Image.open(io.BytesIO(r.content))
+  #Save image
+  file_saveloc=os.path.join(save_loc,f"GenImage_{date.today().strftime('%Y-%m-%d')}_{random_str()}.jpg")
+  image.save(file_saveloc)
+  #Upload image to backblaze
+  _=BB_uploadfile(file_saveloc,os.path.basename(file_saveloc))  
+
+  return image,file_saveloc