Update app.py

app.py

@@ -22,7 +22,7 @@ def get_text_after_colon(input_text):
         # Return the original text if ":" is not found
         return input_text
 
-def infer(image_input, audience):
+def infer(image_input, audience, keyword, protagonist):
     gr.Info('Calling CLIP Interrogator, 이미지를 해석하고 있습니다...')
     clipi_result = clipi_client.predict(
     				image_input,	# str (filepath or URL to image) in 'parameter_3' Image component
@@ -37,8 +37,9 @@ def infer(image_input, audience):
     I'll give you a simple image caption, please provide a fictional story for a {audience} audience that would fit well with the image. Please be creative, do not worry and only generate a cool fictional story. 
     Here's the image description: 
     '{clipi_result[0]}'
+    Keyword: {keyword}
+    Protagonist: {protagonist}
     한국어로 답변해줘.
-    
     """
     gr.Info('Calling ChatGPT, 이야기를 만들고 있습니다...')
     #result = client.predict(
@@ -131,10 +132,13 @@ with gr.Blocks(css=css) as demo:
             <p style="text-align: center">ChatGPT 응답이 오래 지연되거나 사용제한으로 안될 때가 있습니다.</p>
             """
         )
+        
         with gr.Row():
             with gr.Column():
                 image_in = gr.Image(label="이미지 입력", type="filepath", elem_id="image-in", height=420)
                 audience = gr.Radio(label="대상", choices=["Children", "Adult"], value="Children")
+                keyword_in = gr.Textbox(label="핵심 키워드") # 핵심 키워드 입력 상자
+                protagonist_in = gr.Textbox(label="주인공")  # 주인공 입력 상자
                 submit_btn = gr.Button('글을 만들어 주세요')
             with gr.Column():
                 #caption = gr.Textbox(label="Generated Caption")
@@ -151,8 +155,9 @@ with gr.Blocks(css=css) as demo:
                     outputs=[story, share_group],
                     cache_examples=True
                    )
-        
-    submit_btn.click(fn=infer, inputs=[image_in, audience], outputs=[story, share_group])
+
+    submit_btn.click(fn=infer, inputs=[image_in, audience, keyword_in, protagonist_in], outputs=[story, share_group])
+    # submit_btn.click(fn=infer, inputs=[image_in, audience], outputs=[story, share_group])
     share_button.click(None, [], [], _js=share_js)
 
 demo.queue(max_size=12).launch()