Update app.py

app.py

@@ -1,163 +1,78 @@
 import gradio as gr
 from share_btn import community_icon_html, loading_icon_html, share_js
 import re
-import os 
-#hf_token = os.environ.get('HF_TOKEN')
+import os
 import openai
 OPENAI_API_KEY = os.environ.get('OPENAI_API_KEY')
 from gradio_client import Client
-#client = Client("https://fffiloni-test-llama-api-debug.hf.space/", hf_token=hf_token)
 clipi_client = Client("https://fffiloni-clip-interrogator-2.hf.space/")
 
-def get_text_after_colon(input_text):
-    # Find the first occurrence of ":"
-    colon_index = input_text.find(":")
-    
-    # Check if ":" exists in the input_text
-    if colon_index != -1:
-        # Extract the text after the colon
-        result_text = input_text[colon_index + 1:].strip()
-        return result_text
-    else:
-        # Return the original text if ":" is not found
-        return input_text
+def generate_image_url(keywords):
+    return f"https://image.pollinations.ai/prompt/{keywords}"
 
 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
-    				"best",	# str in 'Select mode' Radio component
-    				4,	# int | float (numeric value between 2 and 24) in 'best mode max flavors' Slider component
-    				api_name="/clipi2"
-    )
-    print(clipi_result)
-   
+    clipi_result = clipi_client.predict(image_input, "best", 4, api_name="/clipi2")[0]
 
-    llama_q = f"""
-    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]}'
+    story_intro = f"""
+    # Illustrated Tales
+    ## Created by [Sigkawat Pengnoo](https://flowgpt.com/prompt/qzv2D3OvHkzkfSE4rQCqv) at FlowGPT
+    
     Keyword: {keyword}
     Protagonist: {protagonist}
     한국어로 답변해줘.
-    """
-    gr.Info('Calling ChatGPT, 이야기를 만들고 있습니다...')
-    #result = client.predict(
-    #				llama_q,	# str in 'Message' Textbox component
-    #                "I2S",
-    #				api_name="/predict"
-    #)
-    chat_completion = openai.ChatCompletion.create(model="gpt-3.5-turbo-16k", messages=[{"role": "user", "content": llama_q}])
-    result = chat_completion.choices[0].message.content
-    
-    print(f"Llama2 result: {result}")
 
-    result = get_text_after_colon(result)
+    STORY : "{{ {clipi_result} }}"
 
-    # Split the text into paragraphs based on actual line breaks
-    paragraphs = result.split('\n')
-    
-    # Join the paragraphs back with an extra empty line between each paragraph
-    formatted_text = '\n'.join(paragraphs)
+    Let's begin with Chapter 1!
+    """
 
+    chapters = []
+    current_image_url = generate_image_url(clipi_result)
+    for chapter_num in range(1, 4): # 3개의 장을 생성합니다.
+        chapter_prompt = f"{story_intro}\n\n![Chapter {chapter_num} Image]({current_image_url})\n\nChapter {chapter_num} 내용을 만들어줘."
+        chat_completion = openai.ChatCompletion.create(model="gpt-3.5-turbo-16k", messages=[{"role": "user", "content": chapter_prompt}])
+        chapter_story = chat_completion.choices[0].message.content
+        chapters.append(chapter_story)
+        current_image_url = generate_image_url(chapter_story) # 다음 장의 이미지 URL을 생성합니다.
 
+    formatted_text = '\n'.join(chapters)
     return formatted_text, gr.Group.update(visible=True)
 
-css="""
-#col-container {max-width: 910px; margin-left: auto; margin-right: auto;}
-a {text-decoration-line: underline; font-weight: 600;}
-a {text-decoration-line: underline; font-weight: 600;}
-.animate-spin {
-  animation: spin 1s linear infinite;
-}
-@keyframes spin {
-  from {
-      transform: rotate(0deg);
-  }
-  to {
-      transform: rotate(360deg);
-  }
-}
-#share-btn-container {
-  display: flex; 
-  padding-left: 0.5rem !important; 
-  padding-right: 0.5rem !important; 
-  background-color: #000000; 
-  justify-content: center; 
-  align-items: center; 
-  border-radius: 9999px !important; 
-  max-width: 13rem;
-}
-div#share-btn-container > div {
-    flex-direction: row;
-    background: black;
-    align-items: center;
-}
-#share-btn-container:hover {
-  background-color: #060606;
-}
-#share-btn {
-  all: initial; 
-  color: #ffffff;
-  font-weight: 600; 
-  cursor:pointer; 
-  font-family: 'IBM Plex Sans', sans-serif; 
-  margin-left: 0.5rem !important; 
-  padding-top: 0.5rem !important; 
-  padding-bottom: 0.5rem !important;
-  right:0;
-}
-#share-btn * {
-  all: unset;
-}
-#share-btn-container div:nth-child(-n+2){
-  width: auto !important;
-  min-height: 0px !important;
-}
-#share-btn-container .wrap {
-  display: none !important;
-}
-#share-btn-container.hidden {
-  display: none!important;
-}
+css = """
+# 여기에 이전에 정의한 CSS 코드를 넣으세요.
 """
 
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(
             """
-            <h1 style="text-align: center">Image to Story - Korean</h1>
-            <p style="text-align: center">이미지를 업로드하세요, ChatGPT를 통해 한국어로 이야기를 만들어 줍니다!</p>
-            <p style="text-align: center">원본 https://huggingface.co/spaces/fffiloni/Image-to-Story 에서 한국어로 글을 생성하게 하고, Llama 를 ChatGPT 로 수정한 것입니다.</p>
-            <p style="text-align: center">ChatGPT 응답이 오래 지연되거나 사용제한으로 안될 때가 있습니다.</p>
+            <h1 style="text-align: center">Illustrated Tales - Korean</h1>
+            <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('글을 만들어 주세요')
+                keyword_in = gr.Textbox(label="핵심 키워드")
+                protagonist_in = gr.Textbox(label="주인공")
+                submit_btn = gr.Button('이야기와 그림을 만들어 주세요')
             with gr.Column():
-                #caption = gr.Textbox(label="Generated Caption")
-                story = gr.Textbox(label="생성된 스토리", elem_id="story")
-
+                story = gr.Textbox(label="생성된 이야기와 그림", elem_id="story")
                 with gr.Group(elem_id="share-btn-container", visible=False) as share_group:
                     community_icon = gr.HTML(community_icon_html)
                     loading_icon = gr.HTML(loading_icon_html)
                     share_button = gr.Button("Share to community", elem_id="share-btn")
-        
-        gr.Examples(examples=[["./examples/crabby.png", "Children"],["./examples/hopper.jpeg", "Adult"]],
+
+        gr.Examples(examples=[["./examples/crabby.png", "Children", "해변에서의 모험", "게 주인공"],["./examples/hopper.jpeg", "Adult", "산에서의 여행", "토끼 주인공"]],
                     fn=infer,
-                    inputs=[image_in, audience],
+                    inputs=[image_in, audience, keyword_in, protagonist_in],
                     outputs=[story, share_group],
                     cache_examples=True
                    )
 
     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()