Update app.py

app.py

@@ -10,9 +10,6 @@ logging.basicConfig(level=logging.INFO)
 api_client = Client("http://211.233.58.202:7960/")
 
 def respond(message):
-    if message.endswith('.webp'):
-        return message  # 이미지 파일명만 반환
-    
     logging.info("Received message: %s", message)
     
     try:
@@ -47,6 +44,8 @@ footer {
 }
 """
 
+
+
 # 이미지 생성을 위한 예제 프롬프트
 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"],
@@ -70,6 +69,7 @@ examples = [
     ["A fantasy map of a fictional world, with detailed terrain and cities.", "q19.webp"]
 ]
 
+
 demo = gr.Interface(
     fn=respond,
     inputs=gr.Textbox(label="Enter your prompt for image generation"),
@@ -80,5 +80,15 @@ demo = gr.Interface(
     cache_examples=False  # 예제 캐싱 비활성화
 )
 
+# Examples 컴포넌트 추가
+image_examples = gr.Examples(
+    examples=[[ex[0], f"{ex[1]}"] for ex in examples],
+    inputs=demo.input_components[0],
+    outputs=demo.output_components[0],
+    fn=respond,
+    cache_examples=False,
+    label="Image Examples"
+)
+
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()