add discord login

fns.py

@@ -113,15 +113,9 @@ def refresh():
     # 현재 작업 디렉토리의 폴더 이름들을 튜플로 반환 (특정 폴더 제외)
     return folders_tuple
 
-def dev(a):
-    if a == "Whitehasthewrongonememberwhoisnotavailabletoyoursetupandthechoicesaregiveneveryyearforcharandcroptoyoursetupicontomakethechangesyouareusing":
-        folders = refresh()
-        return "Do", gr.Dropdown(folders, visible=True, interactive=True), gr.File(visible=True, interactive=True), gr.Button(visible=True, interactive=True), gr.Button(visible=True, interactive=True), gr.Button(visible=True, interactive=True), gr.Button(visible=True, interactive=True)
-    elif a == "Whitehasthewrongonememberwhoisnotavailabletoyoursetupandthechoicesaregiveneveryyearforcharandcroptoyoursetupicontomakethechangesyouareusingmake":
-        os.makedirs("/data/cache")
-        return "Permission denied", gr.Dropdown(visible=False, interactive=False), gr.File(visible=False, interactive=False), gr.Button(visible=False, interactive=False), gr.Button(visible=False, interactive=False), gr.Button(visible=False, interactive=False), gr.Button(visible=False, interactive=False)
-    else:
-        return "Permission denied", gr.Dropdown(visible=False, interactive=False), gr.File(visible=False, interactive=False), gr.Button(visible=False, interactive=False), gr.Button(visible=False, interactive=False), gr.Button(visible=False, interactive=False), gr.Button(visible=False, interactive=False)
+def dev():
+    folders = refresh()
+    return "Do", gr.Dropdown(folders, visible=True, interactive=True)
 
 
 def dev2():

generate.py

@@ -433,7 +433,7 @@ def event(event_img, event_radio, event_txt, position):
             f'/data/cache/event_{krtime}_{i1}.jpg',
             quality=70)
         # 이미지 리스트에 추가
-        print(i)
+
         images_list.append(event_img_s)
 
     

login_discord.py

@@ -0,0 +1,69 @@
+import requests
+import gradio as gr
+from urllib.parse import urlparse, parse_qs
+
+def id_check(request: gr.Request):
+    param = str(request.query_params)
+    if not param:
+        return "Login to Discord", gr.Tab(visible=False)
+
+    code = param.split('code=')
+    print(code[1])
+
+    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)
+    return gr.Button(value=username, icon=avatar_url, link=redirect_uri), gr.Tab(visible=False)
+
+
+def get_discord_user_profile(code):
+    # Discord의 API 엔드포인트
+    token_url = "https://discord.com/api/oauth2/token"
+    user_url = "https://discord.com/api/users/@me"
+
+    try:
+        # 토큰 요청을 위한 데이터
+        data = {
+            'client_id': client_id,  # client_id 설정
+            'client_secret': client_secret,  # client_secret 설정
+            'grant_type': 'authorization_code',
+            'code': code,
+            'redirect_uri': redirect_uri  # redirect_uri 설정
+        }
+        headers = {
+            'Content-Type': 'application/x-www-form-urlencoded'
+        }
+
+        # 액세스 토큰 요청
+        token_response = requests.post(token_url, data=data, headers=headers)
+        token_response.raise_for_status()  # 오류 발생시 예외 발생
+        access_token = token_response.json().get('access_token')
+
+        # 사용자 정보 요청
+        auth_headers = {
+            'Authorization': f'Bearer {access_token}'
+        }
+        user_response = requests.get(user_url, headers=auth_headers)
+        user_response.raise_for_status()  # 오류 발생시 예외 발생
+        user_info = user_response.json()
+
+        user_id = user_info.get('id')
+        username = user_info.get('username') + '#' + user_info.get('discriminator')
+        avatar_hash = user_info.get('avatar')
+        avatar_url = f"https://cdn.discordapp.com/avatars/{user_id}/{avatar_hash}.png" if avatar_hash else None
+
+        return user_id, username, avatar_url
+
+    except requests.exceptions.RequestException as e:
+        print("Error to login")
+        # print(str(e))  # 오류 내용 출력을 원한다면 주석 해제
+        return None, None, None  # 오류 발생시 None 반환
+
+# 사용 예:
+client_id = "1239473790732472370"
+client_secret = "C7RLdB4YArSM9jVk8tsznQnKOzwGKwsO"
+# code = "THE_CODE_FROM_REDIRECT_URI"
+redirect_uri = "http://localhost:7860/"
+

main.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import fns
 import generate
+import login_discord
 
 js_func = """
 function refresh() {
@@ -12,15 +13,22 @@ function refresh() {
     }
 }
 """
+with gr.Blocks(theme=gr.themes.Soft()) as demo: # , js=js_func
+    with gr.Column():
+        with gr.Row():
+            gr.Markdown(
+                """
+                # Objektify
+                """)
+            login = gr.Button(size="sm", value="Login to Discord", link="https://discord.com/oauth2/authorize?client_id=1239473790732472370&response_type=code&redirect_uri=http%3A%2F%2Flocalhost%3A7860%2F&scope=identify", scale=0)
 
-with gr.Blocks(theme=gr.themes.Soft(), js=js_func) as demo:
-    with gr.Row():
         gr.Markdown(
             """
-            # Objektify
             ### custom objekt
-            # Input image and Select Options
-            """)
+            ## Input image and Select Options
+            """
+        )
+
     with gr.Tab("Event"):
         with gr.Row():
             with gr.Column():
@@ -31,7 +39,7 @@ with gr.Blocks(theme=gr.themes.Soft(), js=js_func) as demo:
                 event_position = gr.Slider(label="Y position", maximum=100, minimum=0)
                 event_generate = gr.Button(value="Generate")
             with gr.Column():
-                event_output = gr.Gallery(label="output", interactive=False, type="pil", visible=True, scale=10)
+                event_output = gr.Gallery(label="output", interactive=False, type="pil", visible=True, scale=10, object_fit="scale-down")
             event_generate.click(fn=generate.event, inputs=[event_img, event_radio, event_txt, event_position], outputs=event_output)
 
 
@@ -180,22 +188,22 @@ with gr.Blocks(theme=gr.themes.Soft(), js=js_func) as demo:
 
 
 
-    with gr.Tab("Dev"):
-        dev_txt = gr.Textbox(label="Developer Key", visible=True, interactive=True)
-        dev_btn = gr.Button(value="Do and refresh", visible=True, interactive=True)
+    with gr.Tab("Dev", visible=False) as dev:
+
+        dev_btn = gr.Button(value="refresh", visible=True, interactive=True)
         dev_out = gr.Textbox()
-        dev_drop = gr.Dropdown(visible=False, interactive=False)
-        dev_file = gr.File(visible=False, interactive=False)
-        dev_btn2 = gr.Button(visible=False, interactive=False, value="move")
-        dev_btn3 = gr.Button(visible=False, interactive=False, value="zip")
-        dev_btn4 = gr.Button(visible=False, interactive=False, value="download")
-        dev_btn5 = gr.Button(visible=False, interactive=False, value="remove")
+        dev_drop = gr.Dropdown()
+        dev_file = gr.File()
+        dev_btn2 = gr.Button(value="move")
+        dev_btn3 = gr.Button(value="zip")
+        dev_btn4 = gr.Button(value="download")
+        dev_btn5 = gr.Button(value="remove")
 
         dev_btn2.click(fn=fns.dev2, outputs=dev_out)
         dev_btn3.click(fn=fns.dev3, inputs=dev_drop, outputs=dev_out)
         dev_btn4.click(fn=fns.dev4, inputs=dev_drop, outputs=dev_file)
         dev_btn5.click(fn=fns.dev5, inputs=dev_drop, outputs=dev_out)
-        dev_btn.click(fns.dev, dev_txt, [dev_out, dev_drop, dev_file, dev_btn2, dev_btn3, dev_btn4, dev_btn5])
+        dev_btn.click(fns.dev, outputs=[dev_out, dev_drop])
     gr.Markdown(
         """
         ## support email : ssscosmotest@gmail.com
@@ -227,6 +235,9 @@ with gr.Blocks(theme=gr.themes.Soft(), js=js_func) as demo:
                             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])
+
+
 
 if __name__ == "__main__":
     print("http://localhost:7860")

requirements.txt

@@ -2,4 +2,5 @@ gradio
 numpy
 Pillow
 qrcode
-pytz
+pytz
+requests