Update app.py

app.py

@@ -229,11 +229,9 @@ def generate_trump_image(prompt):
         prompt,
         model="black-forest-labs/FLUX.1-dev"
     )
-    path = "trump.png"
-    image.save(path)
     temp_path = f"/tmp/trump_avatar_{uuid.uuid4().hex[:8]}.png"
     image.save(temp_path)
-    return path + f"?v={uuid.uuid4().hex[:6]}"
+    return temp_path
     
 
 def get_trump_emotion():
@@ -246,17 +244,24 @@ def get_trump_emotion():
     else:
         return "Donald Trump neutral expression, on oval office sofa"
 
+def maybe_update_trump_image():
+    global last_mood
+    current_emotion_prompt = get_trump_emotion()
+    if current_emotion_prompt != last_mood:
+        last_mood = current_emotion_prompt
+        path = generate_trump_image(current_emotion_prompt)
+        trump_image.update(value=path) 
 
+        
 def update_trump_image():
     global last_mood
     current_emotion_prompt = get_trump_emotion()
     if current_emotion_prompt != last_mood:
         last_mood = current_emotion_prompt
         img_path = generate_trump_image(current_emotion_prompt)
-        return gr.update(value=img_path)
+        return img_path
     else:
-        return None
-        
+        return gr.skip()
 
 last_mood = "Donald Trump neutral expression, on oval office sofa"
 
@@ -361,6 +366,12 @@ with gr.Blocks(css="""
             top_p
         )
 
+        current_emotion_prompt = get_trump_emotion()
+        if current_emotion_prompt != last_mood:
+            last_mood = current_emotion_prompt
+            img_path = generate_trump_image(current_emotion_prompt)
+            trump_image.update(value=img_path)
+
         history.append({"role": "assistant", "content": assistant_message})
 
         return history, history
@@ -375,19 +386,13 @@ with gr.Blocks(css="""
         bot_respond,
         inputs=[user_message, history, system_message, max_tokens, temperature, top_p],
         outputs=[history, chatbot],
-    ).then(
-    update_trump_image,
-    outputs=[trump_image]
-)
+    )#.then(update_trump_image,outputs=trump_image)
 
     user_message.submit(
         bot_respond,
         inputs=[user_message, history, system_message, max_tokens, temperature, top_p],
         outputs=[history, chatbot],
-    ).then(
-    update_trump_image,
-    outputs=[trump_image]
-)
+    )#.then(update_trump_image, outputs=trump_image)
 
 if __name__ == "__main__":
     demo.launch()