Update app.py

app.py

@@ -158,6 +158,14 @@ def process_model(target_word):
 
     return fig, similar_words_text, dissimilar_words_text
 
+def change_button_state_true():
+    # If the first button is clicked, enable or disable the second button based on its state
+    return gr.update(interactive=True)
+
+def change_button_state_false():
+    # If the first button is clicked, enable or disable the second button based on its state
+    return gr.update(interactive=False)
+
 
 # Gradio 인터페이스 수정
 with gr.Blocks(css=".plot-box {width: 70%; height: 500px;}") as iface:
@@ -167,9 +175,12 @@ with gr.Blocks(css=".plot-box {width: 70%; height: 500px;}") as iface:
     download_nltk_library()
 
     with gr.Row():
+            word_input = gr.Textbox(label="**강조할 단어 입력**", elem_id="input-box", placeholder="단어를 입력하세요", lines=1, interactive=False)
+            with gr.Column(scale=1):
         # 사용자 입력 박스를 강조하기 위해 스타일을 변경
-        word_input = gr.Textbox(label="**강조할 단어 입력**", elem_id="input-box", placeholder="단어를 입력하세요", lines=1)
-        submit_btn = gr.Button("제출", elem_id="submit-btn")
+            # word_input = gr.Textbox(label="**강조할 단어 입력**", elem_id="input-box", placeholder="단어를 입력하세요", lines=1)
+                load_btn = gr.Button("모델 로링", elem_id="submit-btn")
+                submit_btn = gr.Button("단어 입력", elem_id="submit-btn", interactive=False)
 
     with gr.Row():
         # 시각화 화면의 크기를 CSS로 증가
@@ -179,11 +190,29 @@ with gr.Blocks(css=".plot-box {width: 70%; height: 500px;}") as iface:
             similar_words_output = gr.Textbox(label="유사한 단어", interactive=False, lines=5)
             dissimilar_words_output = gr.Textbox(label="유사하지 않은 단어", interactive=False, lines=5)
 
+    load_btn.click(
+        fn=process_model,
+        inputs=[word_input],
+        outputs=[plot_output, similar_words_output, dissimilar_words_output]  
+    )
+    load_btn.click(
+        fn=change_button_state_true,
+        outputs=submit_btn
+    )
+    load_btn.click(
+        fn=change_button_state_true,
+        outputs=word_input
+    )
     submit_btn.click(
         fn=process_model,
         inputs=[word_input],
         outputs=[plot_output, similar_words_output, dissimilar_words_output]
     )
+    submit_btn.click(
+        fn=change_button_state_false,
+        outputs=load_btn
+    )
+
 
 if __name__ == "__main__":
     iface.launch()