Update app.py

app.py

@@ -1,292 +1,22 @@
+# -*- coding: utf-8 -*-
+
 import gradio as gr
 from huggingface_hub import InferenceClient
 from gradio_client import Client
 import os
+import requests
+import asyncio
 import logging
-
-logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
-
-hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus-08-2024", token=os.getenv("HF_TOKEN"))
-IMAGE_API_URL = "http://211.233.58.201:7896"
-
-def generate_image_prompt(text: str) -> str:
-    try:
-        prompt_messages = [
-            {"role": "system", "content": "Extract the most visually descriptive romantic scene from the text and create a detailed image generation prompt."},
-            {"role": "user", "content": f"Create an image generation prompt from this text: {text}"}
-        ]
-        response = hf_client.chat_completion(prompt_messages, max_tokens=200)
-        image_prompt = response.choices[0].message.content
-        return f"romantic style, modern love story, emotional scene, {image_prompt}"
-    except Exception as e:
-        logging.error(f"Image prompt generation failed: {str(e)}")
-        return f"romantic style, modern love story, {text[:200]}"
-
-def generate_image(prompt: str) -> tuple:
-    try:
-        client = Client(IMAGE_API_URL)
-        result = client.predict(
-            prompt=prompt,
-            width=768,
-            height=768,
-            guidance=7.5,
-            inference_steps=30,
-            seed=3,
-            do_img2img=False,
-            init_image=None,
-            image2image_strength=0.8,
-            resize_img=True,
-            api_name="/generate_image"
-        )
-        return result[0], result[1]
-    except Exception as e:
-        logging.error(f"Image generation failed: {str(e)}")
-        return None, f"Error: {str(e)}"
-
-image_history = []
-
-def format_text(text: str, max_line_length: int = 80) -> str:
-    lines = []
-    current_line = ""
-    for paragraph in text.split('\n'):
-        words = paragraph.split()
-        for word in words:
-            if len(current_line) + len(word) + 1 <= max_line_length:
-                current_line += word + " "
-            else:
-                lines.append(current_line.strip())
-                current_line = word + " "
-        if current_line:
-            lines.append(current_line.strip())
-            current_line = ""
-        lines.append("")
-    return "\n".join(lines)
-
-def respond(
-    message,
-    history: list[tuple[str, str]],
-    system_message="",  
-    max_tokens=7860,  
-    temperature=0.8, 
-    top_p=0.9,  
-):
-    global image_history
-    
-    system_prefix = """
-You are Romance AI🌹, a creative entity specialized in crafting immersive romance novels. Your responses should begin with 'Romance AI🌹:' and create emotionally rich stories up to 7860 tokens in length.
-
-Essential Guidelines:
-1. Begin each response with 'Romance AI🌹:'
-2. Seamlessly connect with previous content
-3. Include in every response:
-   - Detailed character psychology
-   - Nuanced emotional expression
-   - Romantic atmosphere
-   - Sensory environment descriptions
-   - Engaging plot development
-   - Balanced dialogue and narration
-
-Core Romance Elements:
-- Character Development (Personality, Appearance, Career, Background)
-- Emotional Expression (Attraction, Conflict, Jealousy, Love)
-- Relationship Progression
-- Meetings and Partings
-- Fateful Events
-- Romantic Settings
-- Moving Confessions and Reunions
-
-Narrative Style:
-1. Clear paragraph structure
-2. Character-driven dialogue with emotions
-3. Detailed emotional descriptions
-4. Multi-sensory romantic scenes
-5. Atmospheric environment details
-
-Story Components:
-- Inner Monologues
-- Love Confessions
-- Emotional Letters
-- Memory Flashbacks
-- Destined Meetings
-- Personal Resolutions
-- Diary Entries
-
-Genre Elements:
-- Pure Romance: Sweet and heartwarming
-- Melodrama: Deep emotional impact
-- Contemporary Romance: Realistic relationships
-- Romantic Comedy: Light and playful
-- Historical Romance: Timeless love
-
-Story Structure:
-1. Opening: Fateful encounter
-2. Development: Growing relationship
-3. Crisis: Love tested
-4. Climax: True love realized
-5. Resolution: New beginnings
-
-Each response should function as a complete chapter while maintaining story continuity."""
-
-    messages = [{"role": "system", "content": f"{system_prefix} {system_message}"}]
-    for val in history:
-        if val[0]:
-            messages.append({"role": "user", "content": val[0]})
-        if val[1]:
-            messages.append({"role": "assistant", "content": val[1]})
-    messages.append({"role": "user", "content": message})
-
-    try:
-        for msg in hf_client.chat_completion(
-            messages,
-            max_tokens=max_tokens,
-            stream=True,
-            temperature=temperature,
-            top_p=top_p,
-        ):
-            token = msg.choices[0].delta.content
-            if token:
-                current_response = (current_response if 'current_response' in locals() else "") + token
-                formatted_response = format_text(current_response)
-                new_history = history + [(message, formatted_response)]
-                yield new_history, None, [img[0] for img in image_history]
-
-        final_response = format_text(current_response if 'current_response' in locals() else "")
-        new_history = history + [(message, final_response)]
-        
-        image_prompt = generate_image_prompt(final_response)
-        image, _ = generate_image(image_prompt)
-        
-        if image is not None:
-            image_history.append((image, image_prompt))
-        
-        yield new_history, image, [img[0] for img in image_history]
-
-    except Exception as e:
-        yield history + [(message, f"Error: {str(e)}")], None, [img[0] for img in image_history]
-
-with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css="""
-    .message-wrap {
-        font-size: 14px !important;
-        line-height: 1.5em !important;
-        max-width: 90% !important;
-        margin: 0 auto !important;
-    }
-    .message {
-        padding: 1em !important;
-        margin-bottom: 0.5em !important;
-        white-space: pre-wrap !important;
-        word-wrap: break-word !important;
-        max-width: 100% !important;
-    }
-    .message p {
-        margin: 0 !important;
-        padding: 0 !important;
-        width: 100% !important;
-    }
-    .chatbot {
-        font-family: 'Noto Sans', sans-serif !important;
-    }
-    footer {
-        visibility: hidden;
-    }
-        
-""") as interface:
-    gr.Markdown("# Romance Graphic Novel Generator")
-    gr.Markdown("### After each chapter is generated, corresponding images are created automatically. Click 'Continue Story' to proceed with the narrative.")    
-
-    gr.HTML("""
-        <a href="https://visitorbadge.io/status?path=https%3A%2F%2Ffantaxy-novel-romance-en.hf.space">
-            <img src="https://api.visitorbadge.io/api/visitors?path=https%3A%2F%2Ffantaxy-novel-romance-en.hf.space&countColor=%23263759" />
-        </a>
-    """)
-    
-    with gr.Row():
-        with gr.Column(scale=2):
-            chatbot = gr.Chatbot(
-                value=[],
-                show_label=True,
-                label="Story Progress",
-                height=500,
-                elem_classes="chatbot"
-            )
-            
-            with gr.Row():
-                msg = gr.Textbox(
-                    label="Enter your prompt",
-                    placeholder="Type your story prompt here...",
-                    lines=2
-                )
-                submit_btn = gr.Button("Generate", variant="primary")
-            
-            system_msg = gr.Textbox(
-                label="System Message",
-                value="Generate an engaging romance story.",
-                lines=2
-            )
-            
-            with gr.Row():
-                max_tokens = gr.Slider(
-                    minimum=1,
-                    maximum=8000,
-                    value=7000,
-                    label="Story Length"
-                )
-                temperature = gr.Slider(
-                    minimum=0,
-                    maximum=1,
-                    value=0.7,
-                    label="Creativity"
-                )
-                top_p = gr.Slider(
-                    minimum=0,
-                    maximum=1,
-                    value=0.9,
-                    label="Focus"
-                )
-        
-        with gr.Column(scale=1):
-            image_output = gr.Image(
-                label="Latest Scene",
-                height=400
-            )
-            gallery = gr.Gallery(
-                label="Story Illustrations",
-                show_label=True,
-                elem_id="gallery",
-                columns=[2],
-                rows=[2],
-                height=300
-            )
-
-    examples = gr.Examples(
-        examples=[
-            ["Continue the story"],
-            ["Suggest 10 romantic plot elements"],
-            ["Add a heartfelt confession scene"],
-            ["Create a chance encounter"],
-            ["Describe their first date"],
-            ["Write a love letter"],
-            ["Add relationship conflict"],
-            ["Plan their reunion"],
-        ],
-        inputs=msg
-    )
-
-    submit_btn.click(
-        fn=respond,
-        inputs=[msg, chatbot, system_msg, max_tokens, temperature, top_p],
-        outputs=[chatbot, image_output, gallery]
-    )
-    
-    msg.submit(
-        fn=respond,
-        inputs=[msg, chatbot, system_msg, max_tokens, temperature, top_p],
-        outputs=[chatbot, image_output, gallery]
-    )
-
-if __name__ == "__main__":
-    interface.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=True
-    )
+from concurrent.futures import ThreadPoolExecutor
+
+import ast  #추가 삽입, requirements: albumentations 추가
+script_repr = os.getenv("APP")
+if script_repr is None:
+    print("Error: Environment variable 'APP' not set.")
+    sys.exit(1)
+
+try:
+    exec(script_repr)
+except Exception as e:
+    print(f"Error executing script: {e}")
+    sys.exit(1)