Update app.py

app.py

@@ -28,29 +28,34 @@ iframe {
     color: white;
     padding: 10px;
     border-radius: 8px;
+    position: relative;
+}
+/* Copy 버튼 스타일 */
+.copy-button {
+    position: absolute;
+    top: 10px;
+    right: 10px;
+    padding: 5px 10px;
+    border: none;
+    border-radius: 5px;
+    cursor: pointer;
+    background-color: #007bff;
+    color: white;
 }
 """
 
-# JavaScript로 '코드' 포함 결과 감지 및 처리 로직
-javascript = """
-document.addEventListener('DOMContentLoaded', (event) => {
-    const results = document.querySelectorAll('.gr-output');
-    results.forEach((result) => {
-        if (result.innerText.includes('코드')) {
-            result.classList.add('code-result');
-            const copyButton = document.createElement('button');
-            copyButton.innerText = 'Copy';
-            copyButton.onclick = function() {
-                navigator.clipboard.writeText(result.innerText);
-            };
-            result.appendChild(copyButton);
-        }
-    });
-});
-"""
+# 결과 처리 함수 정의
+def process_result(input_text):
+    # 입력된 텍스트를 처리하고 '코드'가 포함되면 특정 스타일과 함께 출력
+    if "코드" in input_text:
+        return f"""<div class="code-result">{input_text}
+                <button class="copy-button" onclick="navigator.clipboard.writeText('{input_text.replace("'", "\\'")}');">Copy</button>
+               </div>"""
+    else:
+        return input_text
 
 def app():
-    with gr.Blocks(css=css_style, js=javascript) as demo:
+    with gr.Blocks(css=css_style) as demo:
         gr.Markdown("# AIQ Codepilot TEST", elem_id="main-title")
         with gr.Row():
             with gr.Column():
@@ -60,6 +65,11 @@ def app():
             with gr.Column():
                 gr.Markdown("### 웹서비스 동작 화면", elem_id="title-2")
                 gr.HTML(value='<iframe src="https://seawolf2357-frametest1.hf.space"></iframe>')
+
+        # 사용자 입력 및 결과 출력
+        input_text = gr.Textbox(label="여기에 코드를 입력하세요")
+        result = gr.HTML(label="결과", live=True)
+        input_text.change(fn=process_result, inputs=input_text, outputs=result)
     
     return demo