Update app.py

app.py

@@ -3,6 +3,7 @@ import requests
 import os
 import io
 from PIL import Image
+from gradio_client import Client
 
 # Get the API token from environment variable
 API_TOKEN = os.environ.get("HF_API_TOKEN")
@@ -23,19 +24,39 @@ def generate_image(prompt):
     image_response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
     image_bytes = image_response.content
     image = Image.open(io.BytesIO(image_bytes))
+
+    with st.spinner("Resizing Image..."):
+        client_resize = Client("doevent/AnimateLCM-SVD")
+        resized_image = client_resize.predict(image, api_name="/resize_image")
+    
+    with st.spinner("Animating Image..."):
+        client_animate = Client("doevent/AnimateLCM-SVD")
+        result = client_animate.predict(
+            resized_image,
+            0,
+            True,
+            1,
+            5,
+            1,
+            1.5,
+            576,
+            320,
+            20,
+            api_name="/video"
+        )
     
-    return image
+    return result
 
 # Streamlit interface for user inputs and displaying outputs
 st.title("Morpheus - Dreams Generator")
-st.write("Enter your feelings and moments of the day to generate a summarization along with an AI-generated image!")
+st.write("Enter your feelings and moments of the day to generate a summarization along with an AI-generated animated image!")
 
 inputs = st.text_area("Enter your emotions, expressions, best and worst moments of the day:", height=200, value="Today was a mix of emotions. I felt happy in the morning but sad in the evening. The best moment was meeting a friend, and the worst was a stressful meeting.")
 
-if st.button("Generate Summary and Image"):
+if st.button("Generate Summary and Animated Image"):
     summary = generate_text_summary(inputs)
     st.write("Summary:", summary)
     
     with st.spinner("Generating Image..."):
-        image = generate_image(inputs)
-        st.image(image, caption="Generated Image", use_column_width=True)
+        result = generate_image(inputs)
+        st.markdown(f"")