import gradio as gr
import os
from dotenv import load_dotenv
from PIL import Image
import io
import base64
import requests
from datetime import datetime
from pytz import timezone
import numpy as np
load_dotenv()
# 初回訪問のメッセージ
welcome_message = """
<h2>このサービスは初めてご利用ですか？</h2>
<p>以下から選択してください。</p>
"""
# Googleフォームの送信URL
apps_script_url = os.environ["APPS_SCRIPT_URL"]
# ローカライズ用辞書
translations = {
"en": {
"welcome_message": "<h2>Is this your first time using this service?</h2><p>Please select below.</p>",
"visit_choices": ["First time", "Returning"],
"returning_message": "<h2>Thank you for returning!</h2><p>Click the button below to proceed.</p>",
"proceed_button": "Proceed",
"survey_title": "<h2>Please answer the survey</h2><p>Fill out the form below and submit.</p>",
"form": {
"source": "1. Where did you learn about this Space?",
"country": "2. Which country or region do you live in?",
"ai_usage": "3. Have you used AI to generate illustrations?",
"ai_usage_choices": ["Select...", "Frequently", "Sometimes", "Never"],
"drawing_experience": "4. Have you practiced traditional hand-drawn illustrations?",
"drawing_experience_choices": ["Select...", "As a hobby", "For work", "Never"],
"issues": "5. (Optional) Tell us about any challenges you've faced while practicing illustrations.",
"interest": "6. (Optional) What interested you in this Space?",
"contact_info": "7. (Optional) Provide your contact information (e.g., SNS, URL, email)",
"contact_info_placeholder": "e.g., Twitter, portfolio URL, email",
"submit_button": "Submit"
}
},
"ja": {
"welcome_message": "<h2>このサービスは初めてご利用ですか？</h2><p>以下から選択してください。</p>",
"visit_choices": ["初めて利用する", "2回目以降"],
"returning_message": "<h2>再訪ありがとうございます！</h2><p>以下のボタンをクリックして進んでください。</p>",
"proceed_button": "進む",
"survey_title": "<h2>アンケートにご回答ください</h2><p>以下のフォームに入力して送信してください。</p>",
"form": {
"source": "1. このSpaceをどこで知りましたか？",
"country": "2. お住まいの国や地域を教えてください。",
"ai_usage": "3. 生成AIでイラスト生成をしたことがありますか？",
"ai_usage_choices": ["選択してください...", "よくする", "ある", "ない"],
"drawing_experience": "4. 生成AIではない従来の手描きイラストを練習した経験はありますか？",
"drawing_experience_choices": ["選択してください...", "趣味で", "仕事で", "ない"],
"issues": "5. （任意）イラストの練習中に困った経験があれば教えてください",
"interest": "6. （任意）このSpaceに興味を持った理由を教えてください。",
"contact_info": "7. （任意）連絡先（SNS、URL、メールアドレスなど）を教えてください",
"contact_info_placeholder": "例: Twitterアカウント、ポートフォリオURL、メールアドレスなど",
"submit_button": "送信"
}
},
"zh": {
"welcome_message": "<h2>这是您第一次使用此服务吗？</h2><p>请从下面选择。</p>",
"visit_choices": ["第一次", "再次访问"],
"returning_message": "<h2>感谢您的再次访问！</h2><p>请点击下面的按钮继续。</p>",
"proceed_button": "继续",
"survey_title": "<h2>请回答问卷</h2><p>填写以下表格并提交。</p>",
"form": {
"source": "1. 您是从哪里得知此服务的？",
"country": "2. 您居住的国家或地区是？",
"ai_usage": "3. 您是否使用过AI生成插图？",
"ai_usage_choices": ["请选择...", "经常", "偶尔", "从未"],
"drawing_experience": "4. 您是否练习过传统手绘插图？",
"drawing_experience_choices": ["请选择...", "作为爱好", "为了工作", "从未"],
"issues": "5. （可选）在练习插图过程中遇到的挑战是什么？",
"interest": "6. （可选）是什么让您对这个Space感兴趣？",
"contact_info": "7. （可选）提供您的联系方式（如：SNS、网址、电子邮件等）",
"contact_info_placeholder": "例如：Twitter、作品集网址、电子邮件",
"submit_button": "提交"
}
}
}
# 言語選択に応じてローカライズ
def localize(language):
t = translations[language]
return {
"welcome_message": t["welcome_message"],
"visit_choices": t["visit_choices"],
"returning_message": t["returning_message"],
"proceed_button": t["proceed_button"],
"form_html": f"""
<h2>{t['survey_title']}</h2>
<form id="survey-form" action="{apps_script_url}" method="POST" target="hidden_iframe">
<label for="source">{t['form']['source']}</label><br>
<input type="text" name="source" id="source" required><br><br>
<label for="country">{t['form']['country']}</label><br>
<input type="text" name="country" id="country" required><br><br>
<label for="ai_usage">{t['form']['ai_usage']}</label><br>
<select name="ai_usage" id="ai_usage" required>
<option value="" selected disabled>{t['form']['ai_usage_choices'][0]}</option>
{"".join([f'<option value="{choice}">{choice}</option>' for choice in t['form']['ai_usage_choices'][1:]])}
</select><br><br>
<label for="drawing_experience">{t['form']['drawing_experience']}</label><br>
<select name="drawing_experience" id="drawing_experience" required>
<option value="" selected disabled>{t['form']['drawing_experience_choices'][0]}</option>
{"".join([f'<option value="{choice}">{choice}</option>' for choice in t['form']['drawing_experience_choices'][1:]])}
</select><br><br>
<label for="issues">{t['form']['issues']}</label><br>
<textarea name="issues" id="issues"></textarea><br><br>
<label for="interest">{t['form']['interest']}</label><br>
<textarea name="interest" id="interest"></textarea><br><br>
<label for="contact_info">{t['form']['contact_info']}</label><br>
<input type="text" name="contact_info" id="contact_info" placeholder="{t['form']['contact_info_placeholder']}"><br><br>
<button type="submit" id="submit-button">{t['form']['submit_button']}</button>
</form>
<iframe name="hidden_iframe" style="display:none;"></iframe>
"""
}
# 初回訪問の選択肢に応じた処理
def handle_visit_choice(choice, language):
if choice == localize(language)["visit_choices"][0]:
return gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)
else:
return gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)
# フォーム送信後の処理
def handle_form_submission(flag_value):
print("form submitted")
return gr.update(visible=False), gr.update(visible=True)
# 進むボタンを押した後の処理
def handle_proceed():
print("more than once")
return gr.update(visible=False), gr.update(visible=True)
script = """
() => {
function attachFormListener() {
const form = document.getElementById("survey-form");
if (form) {
function submitForm() {
console.log('form submitted');
const flagInput = document.querySelector("#form_flag textarea");
flagInput.value = 'true';
flagInput.dispatchEvent(new Event('input'));
}
// イベントを削除
form.removeEventListener("submit", submitForm);
// イベントを追加
form.addEventListener("submit", submitForm);
}
}
// 初期ロード時にイベントリスナーを設定
attachFormListener();
// DOMが動的に更新された場合にもイベントリスナーを再設定
const observer = new MutationObserver(() => {
attachFormListener();
});
observer.observe(document.body, { childList: true, subtree: true });
}
"""
# Google Apps ScriptのURL
feedback_script_url = os.environ["FEEDBACK_SCRIPT_URL"]
# 画像生成パラメータ
def generate_image(mode, weight1, weight2):
# ダミーの画像生成処理
image = Image.new("RGB", (256, 256), color=(int(255*weight1), int(255*weight2), int(255*weight1*weight2)))
return image
# Google Driveに保存する処理
def send_feedback(image, filename):
# numpy.ndarray を PIL.Image に変換
if isinstance(image, np.ndarray):
image = Image.fromarray(image)
# 画像をBase64にエンコード
buffered = io.BytesIO()
image.save(buffered, format="PNG")
img_str = base64.b64encode(buffered.getvalue()).decode()
# Google Apps Scriptに送信
response = requests.post(
feedback_script_url,
json={"image": img_str, "filename": filename}
)
if response.status_code == 200:
return gr.update(value="Thank you for cooperation!/ご協力ありがとうございます！", interactive=False)
else:
return gr.update(value="Failed to send. Try again./送信に失敗しました。もう一度お試しください。", interactive=True)