Update app.py

app.py

@@ -323,8 +323,15 @@ def generate_image_from_prompt(prompt):
         )
 
         if response.status_code == 200:
-            image_content = response.content
-            return image_content
+            # 이미지를 임시 파일로 저장
+            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+            image_path = f"generated_images/image_{timestamp}.png"
+            os.makedirs("generated_images", exist_ok=True)
+            
+            with open(image_path, "wb") as f:
+                f.write(response.content)
+            
+            return image_path  # 파일 경로 반환
         else:
             print(f"Error: {response.status_code}")
             print(f"Response: {response.text}")
@@ -539,9 +546,11 @@ def create_interface():
             state, text, voice_result, text_result, prompt = analyze_voice(audio, current_state)
             return state, text, voice_result, text_result, prompt
 
-        def handle_image_generation(prompt):
-            image_content = generate_image_from_prompt(prompt)
-            return image_content
+def handle_image_generation(prompt):
+    image_path = generate_image_from_prompt(prompt)
+    if image_path and os.path.exists(image_path):
+        return image_path
+    return None
 
         def save_reflection_fixed(text, state):
             if not text.strip():