Update app.py

app.py

@@ -2,11 +2,8 @@ import gradio as gr
 from gradio_client import Client
 import os
 import logging
-import requests
 from PIL import Image
 import io
-import base64
-import traceback
 
 # 로깅 설정
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
@@ -18,44 +15,57 @@ api_client = Client("http://211.233.58.202:7960/")
 # Webhook URL
 WEBHOOK_URL = "https://connect.pabbly.com/workflow/sendwebhookdata/IjU3NjUwNTY0MDYzMTA0MzE1MjZlNTUzYzUxM2Ei_pc"
 
-# Imgur 클라이언트 ID
-IMGUR_CLIENT_ID = "4990123196cbfcc"
-
-def upload_to_imgur(image_data):
+def generate_image(prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
+    logger.info(f"Received message: {prompt}, seed: {seed}, randomize_seed: {randomize_seed}, "
+                 f"width: {width}, height: {height}, guidance_scale: {guidance_scale}, "
+                 f"num_inference_steps: {num_inference_steps}")
+    
     try:
-        logger.debug(f"Image data type: {type(image_data)}")
-        logger.debug(f"Image data length: {len(image_data)}")
+        # 이미지 생성 요청
+        result = api_client.predict(
+            prompt=prompt,
+            seed=seed,
+            randomize_seed=randomize_seed,
+            width=width,
+            height=height,
+            guidance_scale=guidance_scale,
+            num_inference_steps=num_inference_steps,
+            api_name="/infer_t2i"
+        )
+        logger.info("API response received: %s", result)
         
+        # 결과 확인 및 처리
+        if isinstance(result, tuple) and len(result) == 3:
+            image_path, used_seed, translated_prompt = result
+            logger.info(f"Image generated at: {image_path}")
+            logger.info(f"Used seed: {used_seed}")
+            logger.info(f"Translated prompt: {translated_prompt}")
+            
+            # 이미지 파일을 열어서 PIL Image 객체로 변환
+            with Image.open(image_path) as img:
+                # 이미지를 메모리에 저장
+                img_byte_arr = io.BytesIO()
+                img.save(img_byte_arr, format='PNG')
+                img_byte_arr = img_byte_arr.getvalue()
+            
+            return img_byte_arr, used_seed, translated_prompt
+        else:
+            raise ValueError("Unexpected API response format")
+    except Exception as e:
+        logger.error("Error during API request: %s", str(e), exc_info=True)
+        return "Failed to generate image due to an error."
+
+def respond(message, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
+    image_data, used_seed, translated_prompt = generate_image(message, seed, randomize_seed, width, height, guidance_scale, num_inference_steps)
+    
+    if isinstance(image_data, bytes):
         # 이미지 데이터를 PIL Image 객체로 변환
         image = Image.open(io.BytesIO(image_data))
-        
-        # PNG 형식으로 변환
-        buffered = io.BytesIO()
-        image.save(buffered, format="PNG")
-        
-        # Base64로 인코딩
-        im_b64 = base64.b64encode(buffered.getvalue()).decode("utf-8")
-        
-        url = "https://api.imgur.com/3/image"
-        payload = {'image': im_b64, 'type': 'base64'}
-        headers = {'Authorization': f'Client-ID {IMGUR_CLIENT_ID}'}
-        
-        logger.debug("Sending request to Imgur API")
-        response = requests.post(url, headers=headers, data=payload)
-        logger.debug(f"Imgur API response status: {response.status_code}")
-        logger.debug(f"Imgur API response content: {response.text}")
-        
-        response.raise_for_status()
-        
-        return response.json()['data']['link']
-    except Exception as e:
-        logger.error(f"Failed to upload image to Imgur: {str(e)}")
-        logger.error(f"Traceback: {traceback.format_exc()}")
-        if hasattr(e, 'response'):
-            logger.error(f"Response content: {e.response.text}")
-        return None
+        return image, f"Used seed: {used_seed}, Translated prompt: {translated_prompt}"
+    else:
+        return image_data, "Error occurred"
+
 
-# ... (나머지 코드는 그대로 유지)
 
 
 css = """
@@ -92,13 +102,13 @@ examples = [
 
 def use_prompt(prompt):
     return prompt
-
 with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
-
+    gr.Markdown("# Image Generation")
     
     with gr.Row():
         input_text = gr.Textbox(label="Enter your prompt for image generation")
         output_image = gr.Image(label="Generated Image")
+        output_text = gr.Textbox(label="Image Info")
     
     with gr.Row():
         seed = gr.Slider(minimum=0, maximum=1000000, step=1, label="Seed", value=123)
@@ -112,28 +122,20 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
         guidance_scale = gr.Slider(minimum=1, maximum=20, step=0.1, label="Guidance Scale", value=5)
         num_inference_steps = gr.Slider(minimum=1, maximum=100, step=1, label="Number of Inference Steps", value=28)
     
-    with gr.Row():
-        for prompt, image_file in examples:
-            with gr.Column():
-                gr.Image(image_file, label=prompt[:50] + "...")  # 프롬프트의 처음 50자만 표시
-                gr.Button("Use this prompt").click(
-                    fn=use_prompt,
-                    inputs=[],
-                    outputs=input_text,
-                    api_name=False
-                ).then(
-                    lambda x=prompt: x,
-                    inputs=[],
-                    outputs=input_text
-                )
-    
     input_text.submit(
         fn=respond, 
         inputs=[input_text, seed, randomize_seed, width, height, guidance_scale, num_inference_steps], 
-        outputs=output_image
+        outputs=[output_image, output_text]
     )
 
+    # Examples 섹션 추가
+    gr.Examples(
+        examples=[[ex[0]] for ex in examples],  # 프롬프트만 사용
+        inputs=input_text,
+        outputs=[output_image, output_text],
+        fn=respond,
+        cache_examples=False
+    )
 
 if __name__ == "__main__":
-    logger.info(f"Starting application with Imgur Client ID: {IMGUR_CLIENT_ID[:5]}...")
     demo.launch()