Update app.py

app.py

@@ -86,18 +86,23 @@ def close():
 window.close();
 </script>"""
 
+
 @app.route('/close2', methods=['GET', 'POST'])
 def close2():
-    return """<script>if (window.opener) {
+    target_origin = request.args.get('target_origin', '*')  # デフォルトはワイルドカード
+    return f"""<script>
+if (window.opener) {{
   const hashParams = new URLSearchParams(window.location.hash.substring(1));
   const token = hashParams.get("access_token");
-  if (token) {
-    window.opener.postMessage({ token }, "https://soiz1-penguinmod-editor-2.hf.space");
+  if (token) {{
+    window.opener.postMessage({{ token }}, "{target_origin}");
     window.close();
-  }
-}
+  }}
+}}
 window.close();
 </script>"""
 
+
+
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)