Update app.py

app.py

@@ -2,86 +2,31 @@ import gradio as gr
 from gradio_client import Client
 import os
 import logging
-from notion_client import Client as NotionClient
 import requests
-from datetime import datetime
-from PIL import Image
-import io
-import sys
 
 # 로깅 설정
-logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', stream=sys.stdout)
-logger = logging.getLogger(__name__)
-
-try:
-    import gradio as gr
-    from gradio_client import Client
-    import os
-    from notion_client import Client as NotionClient
-    import requests
-    from datetime import datetime
-    from PIL import Image
-    import io
-    logger.info("All modules imported successfully")
-except ImportError as e:
-    logger.error(f"Error importing modules: {e}")
-    sys.exit(1)
+logging.basicConfig(level=logging.INFO)
 
 # API 클라이언트 설정
 api_client = Client("http://211.233.58.202:7960/")
 
-# Notion 클라이언트 설정
-NOTION_TOKEN = os.getenv("NOTION_TOKEN")
-NOTION_DATABASE_ID = "88c9bdadcb2044129af77d5932e1a82a"
-
-def test_notion_connection():
-    logger.info("Starting Notion connection test")
-    if not NOTION_TOKEN:
-        logger.error("NOTION_TOKEN is not set. Please set the environment variable.")
-        return False
+# Webhook URL
+WEBHOOK_URL = "https://connect.pabbly.com/workflow/sendwebhookdata/IjU3NjUwNTY0MDYzMTA0MzE1MjZlNTUzYzUxM2Ei_pc"
 
-    notion = NotionClient(auth=NOTION_TOKEN)
+def send_to_webhook(prompt, image_url):
+    payload = {
+        "prompt": prompt,
+        "image_url": image_url
+    }
     try:
-        logger.debug("Testing Notion connection...")
-        user_info = notion.users.me()
-        logger.info(f"Successfully connected to Notion. User info: {user_info}")
-        
-        logger.debug("Retrieving database info...")
-        db_info = notion.databases.retrieve(database_id=NOTION_DATABASE_ID)
-        logger.info(f"Successfully retrieved database info: {db_info}")
-        
-        return True
-    except Exception as e:
-        logger.error(f"Error connecting to Notion: {str(e)}")
-        if hasattr(e, 'response'):
-            logger.error(f"Notion API response: {e.response.text}")
-        return False
-
-def add_to_notion(prompt, image_path):
-    if not notion:
-        logger.warning("Notion integration is not available. Skipping add_to_notion.")
-        return
-
-    try:
-        logger.debug(f"Adding to Notion - Prompt: {prompt}, Image Path: {image_path}")
-        now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-        
-        new_page = {
-            "Prompt": {"title": [{"text": {"content": prompt}}]},
-            "Image": {"url": f"http://211.233.58.202:7960/file={image_path}"},
-            "Date": {"date": {"start": now}}
-        }
-        logger.debug(f"New page properties: {new_page}")
-        
-        response = notion.pages.create(parent={"database_id": NOTION_DATABASE_ID}, properties=new_page)
-        logger.info(f"Successfully added to Notion: {response}")
-    except Exception as e:
-        logger.error(f"Error adding to Notion: {str(e)}")
-        if hasattr(e, 'response'):
-            logger.error(f"Notion API response: {e.response.text}")
+        response = requests.post(WEBHOOK_URL, json=payload)
+        response.raise_for_status()
+        logging.info(f"Successfully sent data to webhook. Status code: {response.status_code}")
+    except requests.exceptions.RequestException as e:
+        logging.error(f"Failed to send data to webhook: {e}")
 
 def respond(message, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
-    logger.info(f"Received message: {message}, seed: {seed}, randomize_seed: {randomize_seed}, "
+    logging.info(f"Received message: {message}, seed: {seed}, randomize_seed: {randomize_seed}, "
                  f"width: {width}, height: {height}, guidance_scale: {guidance_scale}, "
                  f"num_inference_steps: {num_inference_steps}")
     
@@ -97,37 +42,43 @@ def respond(message, seed, randomize_seed, width, height, guidance_scale, num_in
             num_inference_steps=num_inference_steps,
             api_name="/infer_t2i"
         )
-        logger.info("API response received: %s", result)
+        logging.info("API response received: %s", result)
         
         # 결과 확인 및 처리
-        if isinstance(result, tuple) and len(result) >= 1:
-            image_path = result[0]
-            # Notion에 추가
-            add_to_notion(message, image_path)
-            return Image.open(image_path)
+        if isinstance(result, dict) and 'url' in result:
+            image_url = result['url']
+            # Webhook으로 데이터 전송
+            send_to_webhook(message, image_url)
+            return image_url
+        elif isinstance(result, tuple):
+            logging.error("Unexpected tuple response: %s", result)
+            image_url = result[0]
+            # Webhook으로 데이터 전송
+            send_to_webhook(message, image_url)
+            return image_url
         else:
             raise ValueError("Unexpected API response format")
     except Exception as e:
-        logger.error("Error during API request: %s", str(e))
+        logging.error("Error during API request: %s", str(e))
         return "Failed to generate image due to an error."
 
-
-
-
 css = """
 footer {
     visibility: hidden;
 }
 """
 
+
+
+# 이미지 생성을 위한 예제 프롬프트
 examples = [
-    ["A glamorous young woman with long, wavy blonde hair and smokey eye makeup, posing in a luxury hotel room. She's wearing a sparkly gold cocktail dress and holding up a white card with 'openfree.ai' written on it in elegant calligraphy. Soft, warm lighting creates a luxurious atmosphere. ", "q1.webp"],
+    ["A glamorous young woman with long, wavy blonde hair and smokey eye makeup, posing in a luxury hotel room. She’s wearing a sparkly gold cocktail dress and holding up a white card with 'openfree.ai' written on it in elegant calligraphy. Soft, warm lighting creates a luxurious atmosphere. ", "q1.webp"],
     ["A fit male fitness influencer with short dark hair and stubble, standing shirtless in a modern gym. He has defined abs and arm muscles, and is holding a protein shake in one hand and a card that says 'openfree.ai' in the other. Bright, clean lighting highlights his physique.", "q2.webp"],
-    ["A bohemian-style female travel blogger with sun-kissed skin and messy beach waves, sitting on a tropical beach at sunset. She's wearing a flowy white sundress and holding up a weathered postcard with 'openfree.ai' scrawled on it. Golden hour lighting bathes the scene in warm tones. ", "q3.webp"],
-    ["A trendy male fashion influencer with perfectly styled hair and designer stubble, posing on a city street. He's wearing a tailored suit and holding up a sleek black business card with 'openfree.ai' printed in minimalist white font. The background shows blurred city lights, creating a chic urban atmosphere.", "q4.webp"],
-    ["A fresh-faced young female beauty guru with freckles and natural makeup, sitting at a vanity covered in cosmetics. She's wearing a pastel pink robe and holding up a makeup palette with 'openfree.ai' written on it in lipstick. Soft, flattering lighting enhances her radiant complexion. ", "q5.webp"],
-    ["A stylish young woman with long, wavy ombre hair and winged eyeliner, posing in front of a neon-lit city skyline at night. She's wearing a sleek black leather jacket over a sparkly crop top and holding up a holographic business card that says 'openfree.ai' in futuristic font. The card reflects the colorful neon lights, creating a cyberpunk aesthetic.", "q6.webp"],    
-    ['Create a surreal advertisement poster for a fictional time travel agency. The background should depict a swirling vortex of clock faces and historical landmarks from different eras. In the foreground, place large, bold text that reads "AI TOURS: YOUR PAST IS OUR FUTURE" in a retro-futuristic font. The text should appear to be partially disintegrating into particles that are being sucked into the time vortex. Include smaller text at the bottom with fictional pricing and the slogan "History is just a ticket away!"', "q7.webp"],
+    ["A bohemian-style female travel blogger with sun-kissed skin and messy beach waves, sitting on a tropical beach at sunset. She’s wearing a flowy white sundress and holding up a weathered postcard with 'openfree.ai' scrawled on it. Golden hour lighting bathes the scene in warm tones. ", "q3.webp"],
+    ["A trendy male fashion influencer with perfectly styled hair and designer stubble, posing on a city street. He’s wearing a tailored suit and holding up a sleek black business card with 'openfree.ai' printed in minimalist white font. The background shows blurred city lights, creating a chic urban atmosphere.", "q4.webp"],
+    ["A fresh-faced young female beauty guru with freckles and natural makeup, sitting at a vanity covered in cosmetics. She’s wearing a pastel pink robe and holding up a makeup palette with 'openfree.ai' written on it in lipstick. Soft, flattering lighting enhances her radiant complexion. ", "q5.webp"],
+    ["A stylish young woman with long, wavy ombre hair and winged eyeliner, posing in front of a neon-lit city skyline at night. She’s wearing a sleek black leather jacket over a sparkly crop top and holding up a holographic business card that says 'openfree.ai' in futuristic font. The card reflects the colorful neon lights, creating a cyberpunk aesthetic.", "q6.webp"],    
+    ["Create a surreal advertisement poster for a fictional time travel agency. The background should depict a swirling vortex of clock faces and historical landmarks from different eras. In the foreground, place large, bold text that reads “AI TOURS: YOUR PAST IS OUR FUTURE” in a retro-futuristic font. The text should appear to be partially disintegrating into particles that are being sucked into the time vortex. Include smaller text at the bottom with fictional pricing and the slogan “History is just a ticket away!”", "q7.webp"],
     ["Photo realistic scene inspired by LOTR: [A tiny red dragon sleeps curled up in a nest on a medieval wizard's table]. Shot with a macro lens (f/2.8, 50mm) and a Canon EOSR5, the soft focus captures [the cozy morning light filtering through a near by window]. The pastel colors and whimsical steam shapes enhance the serene atmosphere, evoking a DnD RPG setting. The image is rendered in 16K and 8K, highlighting [the intricate details and medieval charm].", "q8.webp"],
     ["썬글라스를 착용한 귀여운 강아지가 'Openfree.ai'라고 쓰여진 표지판을 들고있다.", "q9.webp"],
     ["미레적인 도시의 해지는 풍경", "q10.webp"],
@@ -143,7 +94,12 @@ examples = [
     ["A fantasy map of a fictional world, with detailed terrain and cities.", "q19.webp"]
 ]
 
+def use_prompt(prompt):
+    return prompt
+
 with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
+
+    
     with gr.Row():
         input_text = gr.Textbox(label="Enter your prompt for image generation")
         output_image = gr.Image(label="Generated Image")
@@ -181,22 +137,5 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
         outputs=output_image
     )
 
-
-
-# ... (나머지 코드는 그대로 유지)
-
 if __name__ == "__main__":
-    logger.info("Starting application")
-    logger.info(f"Notion Database ID: {NOTION_DATABASE_ID}")
-    logger.info(f"Notion Token (first 5 chars): {NOTION_TOKEN[:5] if NOTION_TOKEN else 'Not set'}")
-    
-    notion_connection_successful = test_notion_connection()
-    
-    if notion_connection_successful:
-        logger.info("Notion connection test passed. Starting Gradio interface...")
-        notion = NotionClient(auth=NOTION_TOKEN)
-    else:
-        logger.error("Failed to connect to Notion. The application will continue without Notion integration.")
-        notion = None
-    
     demo.launch()