Update app.py

app.py

@@ -1,38 +1,45 @@
-import gradio as gr
+import streamlit as st
 from huggingface_hub import HfApi
 import os
-import shutil
 
 # Initialize the Hugging Face API with the token
 api = HfApi(token=os.getenv("HF_API_TOKEN"))
 
-def upload_video(video):
-    # Save the uploaded video to a temporary file
-    video_path = os.path.join("temp", video.name)
+# Function to handle video uploads to Hugging Face
+def upload_video_to_hf(video_file, video_name):
+    # Save the video temporarily to a file
+    video_path = os.path.join("temp", video_name)
     
-    # Write the file from the uploaded content
+    # Ensure the temp directory exists
+    if not os.path.exists("temp"):
+        os.makedirs("temp")
+    
+    # Write the video content to a file
     with open(video_path, "wb") as f:
-        shutil.copyfileobj(video, f)  # Use shutil.copyfileobj to write the file
+        f.write(video_file.read())
 
     # Upload the video to Hugging Face
     api.upload_file(
         path_or_fileobj=video_path,
-        path_in_repo=f"/videos/{video.name}",
+        path_in_repo=f"videos/{video_name}",
         repo_id="vericudebuget/ok4231",
         repo_type="space",
     )
-    
-    return f"Uploaded {video.name} successfully!"
 
-def progress_bar(video):
-    return f"Uploading {video.name}..."
+    return f"Uploaded {video_name} successfully to Hugging Face!"
 
-with gr.Blocks() as demo:
-    video_input = gr.File(label="Upload Video")
-    progress = gr.Textbox(label="Progress")
-    output = gr.Textbox(label="Output")
-    
-    video_input.change(progress_bar, inputs=video_input, outputs=progress)
-    video_input.upload(upload_video, inputs=video_input, outputs=output)
+# Streamlit app interface
+st.title("Video Uploader to Hugging Face")
 
-demo.launch()
+# File uploader for video input
+uploaded_video = st.file_uploader("Upload your video", type=["mp4", "avi", "mov"])
+
+if uploaded_video:
+    # Show the name of the uploaded video
+    st.write(f"Uploaded video: {uploaded_video.name}")
+    
+    # Upload the video when the button is clicked
+    if st.button("Upload to Hugging Face"):
+        with st.spinner("Uploading..."):
+            result = upload_video_to_hf(uploaded_video, uploaded_video.name)
+            st.success(result)