Update app.py

app.py

@@ -10,6 +10,9 @@ logging.basicConfig(level=logging.INFO)
 api_client = Client("http://211.233.58.202:7960/")
 
 def respond(message):
+    if message.endswith(('.png', '.jpg', '.jpeg', '.webp')):
+        return message, message  # 프롬프트와 이미지 파일명 반환
+    
     logging.info("Received message: %s", message)
     
     try:
@@ -28,15 +31,15 @@ def respond(message):
         
         # 결과 확인 및 처리
         if isinstance(result, dict) and 'url' in result:
-            return result['url']
+            return message, result['url']
         elif isinstance(result, tuple):
             logging.error("Unexpected tuple response: %s", result)
-            return result[0]
+            return message, result[0]
         else:
             raise ValueError("Unexpected API response format")
     except Exception as e:
         logging.error("Error during API request: %s", str(e))
-        return "Failed to generate image due to an error."
+        return message, "Failed to generate image due to an error."
 
 css = """
 footer {
@@ -67,16 +70,17 @@ examples = [
     ["A fantasy map of a fictional world, with detailed terrain and cities.", "q19.webp"]
 ]
 
-# Gradio 인터페이스 설정
 demo = gr.Interface(
     fn=respond,
     inputs=gr.Textbox(label="Enter your prompt for image generation"),
-    outputs=gr.Image(label="Generated Image"),
+    outputs=[
+        gr.Textbox(label="Prompt"),
+        gr.Image(label="Generated Image")
+    ],
     examples=examples,
     theme="Nymbo/Nymbo_Theme",
     css=css,
-    enable_cache=False  # 캐싱 비활성화
-
+    cache_examples=False  # 예제 캐싱 비활성화
 )
 
 if __name__ == "__main__":