update V1.1.2L Beta

-Add OMA to Beta user
-fix bug

generate.py

@@ -51,6 +51,8 @@ def front_static(img, obj_color, obj_color_picker, ai_num, txt_color, txt_color_
             img.paste(front_side, (0, 0), front_side)
         case "Special":
             img = edimg.paste_img(img, "resource/special_side.png", (0, 0))
+        case "OMA":
+            img = edimg.paste_img(img, "resource/oma_side.png", (1016, 83))
         case "Other color":
             obj_color = obj_color_picker
             front_side = edimg.color("resource/front_side.png", obj_color)
@@ -179,6 +181,8 @@ def back(obj_color, obj_color_picker, ai_num, txt_color, txt_color_picker, group
             img = Image.new("RGB", (1164, 1800), obj_color)
         case "Special":
             img = Image.open('resource/special_back.png')
+        case "OMA":
+            img = Image.open('resource/oma_back.png')
         case "Other color":
             obj_color = obj_color_picker
             img = Image.new("RGB", (1164, 1800), obj_color)

login_discord.py

@@ -5,7 +5,7 @@ from urllib.parse import urlparse, parse_qs
 def id_check(request: gr.Request):
     param = str(request.query_params)
     if not "code=" in param:
-        return "Login to Discord", gr.Tab(visible=False)
+        return "Login to Discord", gr.Tab(visible=False), gr.Radio(["Atom01", "Binary01", "Cream01", "Divine01", "Special", "Other color", "AI Color"], label="Objekt Color", value="Cream01")
 
     code = param.split('code=')
     print(code[1])
@@ -13,11 +13,11 @@ def id_check(request: gr.Request):
     user_id, username, avatar_url = get_discord_user_profile(str(code[1]))
     if user_id == None:
         return "Login to Discord", gr.Tab(visible=False)
-    if user_id == "412827396027187211":
-        return gr.Button(value=username, icon=avatar_url, link=redirect_uri), gr.Tab(visible=True)
+    if user_id == "412827396027187211" or user_id == "676049087300239369":
+        return gr.Button(value=username, icon=avatar_url, link=redirect_uri), gr.Tab(visible=True), gr.Radio(["Atom01", "Binary01", "Cream01", "Divine01", "Special", "OMA", "Other color", "AI Color"], label="Objekt Color", value="Divine01")
     print(user_id)
-    return gr.Button(value=username, icon=avatar_url, link=redirect_uri), gr.Tab(visible=False)
-
+    return gr.Button(value=username, icon=avatar_url, link=redirect_uri), gr.Tab(visible=False), gr.Radio(["Atom01", "Binary01", "Cream01", "Divine01", "Special", "Other color", "AI Color"], label="Objekt Color", value="Cream01")
+#1
 
 def get_discord_user_profile(code):
     # Discord의 API 엔드포인트
@@ -60,7 +60,7 @@ def get_discord_user_profile(code):
     except requests.exceptions.RequestException as e:
         print("Error to login")
         # print(str(e))  # 오류 내용 출력을 원한다면 주석 해제
-        return None, None, None  # 오류 발생시 None 반환
+        return None, None, None, None  # 오류 발생시 None 반환
 
 # 사용 예:
 client_id = "1239473790732472370"

main.py

@@ -235,7 +235,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo: # , js=js_func
                             right_btn, rounded_back], outputs=[preview, ex_img_b])
 
     ex_btn.click(fn=generate.ex_gen, inputs=[ex_img_f, ex_file_f, ex_img_b, ex_gen_file_count, ex_hide_blur, ex_hide_blur_range, watermark], outputs=[ex_img_f_out, ex_img_b_out])
-    demo.load(fn=login_discord.id_check, outputs=[login, dev])
+    demo.load(fn=login_discord.id_check, outputs=[login, dev, obj_color])