Update app.py

app.py

@@ -67,17 +67,23 @@ def generate_metadata(video_name, title, description, uploader, file_location, t
 
 def update_index_file(new_metadata_path):
     temp_dir = "temp_repo"
+    
+    # Remove existing temp directory if it exists
     if os.path.exists(temp_dir):
         shutil.rmtree(temp_dir)
     
     try:
+        # Clone the Hugging Face repo
         subprocess.run(['git', 'clone', 'https://huggingface.co/spaces/vericudebuget/ok4231', temp_dir], check=True)
+        
+        # Find all existing JSON metadata files
         metadata_dir = os.path.join(temp_dir, 'metadata')
         json_files = glob.glob(os.path.join(metadata_dir, '*-index.json'))
         
-        base_url = "huggingface.co/spaces/vericudebuget/ok4231/raw/main/metadata/"
+        base_url = "https://huggingface.co/spaces/vericudebuget/ok4231/raw/main/metadata/"
         paths = []
         
+        # Collect existing metadata files with timestamps
         for f in json_files:
             file_timestamp = datetime.now().isoformat()  # Get the current timestamp
             file_path = f"{base_url}{os.path.basename(f)}"
@@ -92,15 +98,19 @@ def update_index_file(new_metadata_path):
         if not any(entry['url'] == new_full_path for entry in paths):
             paths.append({"url": new_full_path, "timestamp": file_timestamp})
         
-        # Convert the paths list to a proper JSON format
+        # Sort the paths by timestamp in descending order (latest to oldest)
+        paths.sort(key=lambda x: x['timestamp'], reverse=True)
+        
+        # Convert the paths list to a JSON format
         index_content = json.dumps(paths, indent=2)
         
+        # Write the sorted index to 'video-index.json'
         index_path = os.path.join(temp_dir, 'metadata', 'video-index.json')
         os.makedirs(os.path.dirname(index_path), exist_ok=True)
         with open(index_path, 'w') as f:
             f.write(index_content)
         
-        # Upload the updated index file
+        # Upload the updated index file to the Hugging Face space
         hf_api.upload_file(
             path_or_fileobj=index_path,
             path_in_repo="metadata/video-index.json",
@@ -109,6 +119,7 @@ def update_index_file(new_metadata_path):
         )
     
     finally:
+        # Clean up by removing the temp directory
         if os.path.exists(temp_dir):
             shutil.rmtree(temp_dir)
 
@@ -119,6 +130,11 @@ def create_subtitles(video_path):  # Renamed from generate_subtitles
         audio = audio.set_channels(1).set_frame_rate(44100).set_sample_width(2)
         audio.export(temp_audio.name, format='mp3', bitrate='128k')
         
+        # Check if the audio is silent
+        if audio.dBFS < -90:  # Adjust the threshold as needed
+            os.unlink(temp_audio.name)  # Clean up temp file
+            return "No audio detected in the video."
+
         # Generate subtitles using Groq
         with open(temp_audio.name, 'rb') as audio_file:
             translation = groq_client.audio.translations.create(
@@ -143,6 +159,7 @@ def create_subtitles(video_path):  # Renamed from generate_subtitles
     os.unlink(temp_audio.name)  # Clean up temp file
     return vtt_content
 
+
 def upload_video_to_hf(video_file, original_video_name, title, description, uploader, should_generate_subs=False, custom_thumbnail=None):  # Renamed parameter
     temp_dir = "temp"
     if not os.path.exists(temp_dir):
@@ -252,9 +269,15 @@ if uploaded_video:
         os.unlink(temp_video.name)  # Clean up temp file
         
         # Subtitle generation toggle, disabled if video is longer than 1 hour
-        should_generate_subs = st.toggle("Generate Subtitles", disabled=video_duration > 3600, value=True)  # Renamed variable
+        should_generate_subs = st.toggle("Generate Subtitles", disabled=video_duration > 7200, value=True)  # Renamed variable
+        if should_generate_subs:
+            st.warning("Subtitles will be generated automatically, and if the video is in another language, it will be translated automatically into English.")
+        if video_duration > 1180:
+            st.warning("Hey there! Just wanted to warn you that uploading pirated movies is not allowed.")
         if video_duration > 3600 and should_generate_subs:
-            st.warning("Subtitle generation is disabled for videos longer than 1 hour.")
+            st.warning("Warning, for videos longer than an hour, generating subtitles will take some time! Please wait :)")
+        if video_duration > 7000:
+            st.warning("Now that's a long video. It will take a long time to upload. Make sure you have the right uploader details!")
         
         custom_thumbnail = st.file_uploader("Upload custom thumbnail (optional)", type=["jpg", "jpeg", "png"])