Update app.py

app.py

@@ -82,33 +82,39 @@ def stop_generation():
     return "생성이 중단되었습니다."
 
 with gr.Blocks() as demo:
-    chatbot = gr.Chatbot()
-    msg = gr.Textbox(label="메시지 입력")
-
+    gr.Markdown("# 프롬프트 최적화 Playground")
+    
     with gr.Row():
-        send = gr.Button("전송")
-        continue_btn = gr.Button("계속 작성")
-        stop = gr.Button("🛑 생성 중단")
-        clear = gr.Button("🗑️ 대화 내역 지우기")
-
-    with gr.Accordion("추가 설정", open=True):
-        system_message = gr.Textbox(
-            value="너는 나의 최고의 비서이다.\n내가 요구하는것들을 최대한 자세하고 정확하게 답변하라.\n반드시 한글로 답변할것.",
-            label="시스템 메시지",
-            lines=5
-        )
-        max_tokens = gr.Slider(minimum=1, maximum=2000, value=500, step=100, label="최대 새 토큰 수")
-        temperature = gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.05, label="온도")
-        top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.90, step=0.05, label="Top-p (핵 샘플링)")
-        model = gr.Radio(list(models.keys()), value=list(models.keys())[0], label="언어 모델 선택", info="사용할 언어 모델을 선택하세요")
+        with gr.Column(scale=1):
+            system_message = gr.Textbox(
+                value="너는 나의 최고의 비서이다.\n내가 요구하는것들을 최대한 자세하고 정확하게 답변하라.\n반드시 한글로 답변할것.",
+                label="시스템 메시지",
+                lines=5
+            )
+            prompt = gr.Textbox(label="프롬프트 입력", lines=3)
+            
+            with gr.Row():
+                send = gr.Button("전송")
+                continue_btn = gr.Button("계속 작성")
+                stop = gr.Button("🛑 생성 중단")
+                clear = gr.Button("🗑️ 대화 내역 지우기")
+            
+            with gr.Accordion("모델 설정", open=True):
+                model = gr.Radio(list(models.keys()), value=list(models.keys())[0], label="언어 모델 선택", info="사용할 언어 모델을 선택하세요")
+                max_tokens = gr.Slider(minimum=1, maximum=2000, value=500, step=100, label="최대 새 토큰 수")
+                temperature = gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.05, label="온도")
+                top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.90, step=0.05, label="Top-p (핵 샘플링)")
+        
+        with gr.Column(scale=2):
+            chatbot = gr.Chatbot(height=700, label="대화 결과")
 
     # Event handlers
-    send.click(respond, inputs=[msg, system_message, max_tokens, temperature, top_p, model], outputs=[chatbot])
-    msg.submit(respond, inputs=[msg, system_message, max_tokens, temperature, top_p, model], outputs=[chatbot])
+    send.click(respond, inputs=[prompt, system_message, max_tokens, temperature, top_p, model], outputs=[chatbot])
+    prompt.submit(respond, inputs=[prompt, system_message, max_tokens, temperature, top_p, model], outputs=[chatbot])
     continue_btn.click(continue_writing, 
                        inputs=[chatbot, system_message, max_tokens, temperature, top_p, model], 
                        outputs=[chatbot])
-    stop.click(stop_generation, outputs=[msg])
+    stop.click(stop_generation, outputs=[prompt])
     clear.click(lambda: None, outputs=[chatbot])
 
 if __name__ == "__main__":