Update app.py

app.py

@@ -30,6 +30,16 @@ from gradio_client import Client
 from urllib.parse import quote
 
 
+# Add these imports at the top of your file
+from gradio_client import Client, handle_file
+import tempfile
+from PIL import Image
+import io
+
+
+
+
+
 # 🎭 App Configuration - Because every app needs a good costume!
 Site_Name = '🐙GitCosmos🌌 - AI Azure Cosmos DB and Github Agent'
 title = "🐙GitCosmos🌌 - AI Azure Cosmos DB and Github Agent"
@@ -797,6 +807,107 @@ def get_video_html(video_path, width="100%"):
     </video>
     '''
 
+
+
+
+# *********
+
+
+
+
+
+def generate_video_from_image(image_data, seed=None, motion_bucket_id=127, fps_id=6):
+    """Generate video from image using Stable Video Diffusion"""
+    try:
+        # Create a temporary file to save the uploaded image
+        with tempfile.NamedTemporaryFile(suffix='.png', delete=False) as temp_img:
+            # If image_data is a PIL Image, convert to bytes
+            if isinstance(image_data, Image.Image):
+                img_byte_arr = io.BytesIO()
+                image_data.save(img_byte_arr, format='PNG')
+                temp_img.write(img_byte_arr.getvalue())
+            else:
+                temp_img.write(image_data)
+            temp_img_path = temp_img.name
+
+        # Initialize the Gradio client
+        client = Client("awacke1/stable-video-diffusion")
+
+        # Get random seed if none provided
+        if seed is None:
+            seed = client.predict(api_name="/get_random_value")
+
+        # Generate video
+        result = client.predict(
+            image=temp_img_path,
+            seed=seed,
+            randomize_seed=True,
+            motion_bucket_id=motion_bucket_id,
+            fps_id=fps_id,
+            api_name="/video"
+        )
+
+        # result[0] contains Dict with video path
+        video_path = result[0]['video']
+        return video_path, result[1]  # Return video path and used seed
+
+    except Exception as e:
+        st.error(f"Error generating video: {str(e)}")
+        return None, None
+
+# Add this to the 'Show as Run AI' section in your main function, 
+# right after the "🤖 Run AI" button:
+
+# Add image upload and video generation
+st.image_uploader = st.file_uploader("Upload Image for Video Generation 🖼️", type=['png', 'jpg', 'jpeg'])
+st.video_gen_params = {
+    'motion_bucket_id': st.slider("Motion Intensity 🌊", 1, 255, 127),
+    'fps_id': st.slider("Frames per Second 🎬", 1, 30, 6)
+}
+
+if st.image_uploader is not None:
+    if st.button("🎥 Generate Video"):
+        with st.spinner("Generating video... 🎬"):
+            # Read uploaded image
+            image_bytes = st.image_uploader.read()
+            
+            # Generate video
+            video_path, used_seed = generate_video_from_image(
+                image_bytes,
+                motion_bucket_id=st.video_gen_params['motion_bucket_id'],
+                fps_id=st.video_gen_params['fps_id']
+            )
+            
+            if video_path:
+                # Save video to local storage
+                video_filename = f"generated_video_{datetime.now().strftime('%Y%m%d_%H%M%S')}.mp4"
+                shutil.copy(video_path, video_filename)
+                
+                st.success(f"Video generated successfully! Seed used: {used_seed}")
+                
+                # Display the generated video
+                st.video(video_filename)
+                
+                # Save to Cosmos DB if needed
+                if container:
+                    video_record = {
+                        "id": generate_unique_id(),
+                        "type": "generated_video",
+                        "filename": video_filename,
+                        "seed": used_seed,
+                        "motion_bucket_id": st.video_gen_params['motion_bucket_id'],
+                        "fps_id": st.video_gen_params['fps_id'],
+                        "timestamp": datetime.now().isoformat()
+                    }
+                    success, message = insert_record(container, video_record)
+                    if success:
+                        st.success("Video record saved to database")
+                    else:
+                        st.error(f"Error saving video record: {message}")
+
+
+# ******************************************
+
 # Function to create HTML for audio player (when life needs a soundtrack 🎶)
 def get_audio_html(audio_path, width="100%"):
     audio_url = f"data:audio/mpeg;base64,{base64.b64encode(open(audio_path, 'rb').read()).decode()}"