add and disable rumble option, webscraper is buggin

data/rumble/subtitles.srt

@@ -0,0 +1,404 @@
+1.1
+00:00:00,320 --> 00:00:00,380
+and
+
+1.2
+00:00:00,380 --> 00:00:00,600
+get
+
+1.3
+00:00:00,600 --> 00:00:00,780
+to
+
+1.4
+00:00:00,780 --> 00:00:00,880
+the
+
+1.5
+00:00:00,880 --> 00:00:01,340
+core
+
+1.6
+00:00:01,340 --> 00:00:01,580
+of
+
+1.7
+00:00:01,580 --> 00:00:01,760
+the
+
+1.8
+00:00:01,760 --> 00:00:02,140
+issue,
+
+1.9
+00:00:02,379 --> 00:00:02,480
+to
+
+1.10
+00:00:02,480 --> 00:00:02,799
+get
+
+1.11
+00:00:02,799 --> 00:00:02,980
+to
+
+1.12
+00:00:02,980 --> 00:00:03,080
+the
+
+1.13
+00:00:03,080 --> 00:00:03,740
+fundamental
+
+1.14
+00:00:03,740 --> 00:00:04,320
+point,
+
+1.15
+00:00:05,059 --> 00:00:05,799
+which
+
+1.16
+00:00:05,799 --> 00:00:05,960
+is
+
+1.17
+00:00:05,960 --> 00:00:06,240
+why
+
+1.18
+00:00:06,240 --> 00:00:06,419
+a
+
+1.19
+00:00:06,419 --> 00:00:06,559
+lot
+
+1.20
+00:00:06,559 --> 00:00:06,679
+of
+
+1.21
+00:00:06,679 --> 00:00:06,839
+people
+
+1.22
+00:00:06,839 --> 00:00:07,099
+don't
+
+1.23
+00:00:07,099 --> 00:00:07,299
+like
+
+1.24
+00:00:07,299 --> 00:00:07,519
+me
+
+1.25
+00:00:07,519 --> 00:00:07,780
+because
+
+1.26
+00:00:07,780 --> 00:00:07,980
+it's
+
+2.1
+00:00:07,980 --> 00:00:08,679
+brash
+
+2.2
+00:00:08,679 --> 00:00:08,939
+and
+
+2.3
+00:00:08,939 --> 00:00:09,099
+you
+
+2.4
+00:00:09,099 --> 00:00:09,220
+know,
+
+2.5
+00:00:09,339 --> 00:00:09,359
+when
+
+2.6
+00:00:09,359 --> 00:00:09,460
+you
+
+2.7
+00:00:09,460 --> 00:00:09,939
+go
+
+2.8
+00:00:09,939 --> 00:00:10,199
+right
+
+2.9
+00:00:10,199 --> 00:00:10,380
+with
+
+2.10
+00:00:10,380 --> 00:00:10,519
+the
+
+2.11
+00:00:10,519 --> 00:00:10,919
+fundamental
+
+2.12
+00:00:10,919 --> 00:00:11,380
+point,
+
+2.13
+00:00:11,759 --> 00:00:12,439
+I
+
+2.14
+00:00:12,439 --> 00:00:12,640
+think
+
+2.15
+00:00:12,640 --> 00:00:13,000
+a
+
+2.16
+00:00:13,000 --> 00:00:13,160
+lot
+
+2.17
+00:00:13,160 --> 00:00:13,300
+of
+
+2.18
+00:00:13,300 --> 00:00:13,539
+people
+
+2.19
+00:00:13,539 --> 00:00:13,779
+are
+
+2.20
+00:00:13,779 --> 00:00:14,080
+ready
+
+2.21
+00:00:14,080 --> 00:00:14,320
+for
+
+2.22
+00:00:14,320 --> 00:00:14,580
+that,
+
+2.23
+00:00:14,699 --> 00:00:14,820
+you
+
+2.24
+00:00:14,820 --> 00:00:15,000
+know,
+
+2.25
+00:00:15,220 --> 00:00:15,339
+if
+
+2.26
+00:00:15,339 --> 00:00:15,740
+they're
+
+2.27
+00:00:15,740 --> 00:00:15,859
+in
+
+2.28
+00:00:15,859 --> 00:00:16,000
+a
+
+2.29
+00:00:16,000 --> 00:00:16,300
+totally
+
+3.1
+00:00:16,300 --> 00:00:16,699
+if
+
+3.2
+00:00:16,699 --> 00:00:16,879
+they
+
+3.3
+00:00:16,879 --> 00:00:17,000
+have
+
+3.4
+00:00:17,000 --> 00:00:17,079
+a
+
+3.5
+00:00:17,079 --> 00:00:17,320
+totally
+
+3.6
+00:00:17,320 --> 00:00:17,640
+different
+
+3.7
+00:00:17,640 --> 00:00:18,239
+perspective.
+
+3.8
+00:00:18,679 --> 00:00:18,739
+What
+
+3.9
+00:00:18,739 --> 00:00:18,839
+I
+
+3.10
+00:00:18,839 --> 00:00:18,980
+mean
+
+3.11
+00:00:18,980 --> 00:00:19,160
+by
+
+3.12
+00:00:19,160 --> 00:00:19,339
+that
+
+3.13
+00:00:19,339 --> 00:00:19,460
+is
+
+3.14
+00:00:19,460 --> 00:00:19,600
+when
+
+3.15
+00:00:19,600 --> 00:00:19,719
+I
+
+3.16
+00:00:19,719 --> 00:00:19,899
+go
+
+3.17
+00:00:19,899 --> 00:00:20,079
+out
+
+3.18
+00:00:20,079 --> 00:00:20,280
+there
+
+3.19
+00:00:20,280 --> 00:00:20,399
+and
+
+3.20
+00:00:20,399 --> 00:00:20,559
+say,
+
+3.21
+00:00:20,660 --> 00:00:20,820
+hey,
+
+3.22
+00:00:21,100 --> 00:00:21,480
+Christ
+
+3.23
+00:00:21,480 --> 00:00:21,739
+is
+
+3.24
+00:00:21,739 --> 00:00:22,059
+king
+
+3.25
+00:00:22,059 --> 00:00:22,620
+and
+
+3.26
+00:00:22,620 --> 00:00:22,820
+that
+
+3.27
+00:00:22,820 --> 00:00:23,059
+means
+
+4.1
+00:00:23,300 --> 00:00:23,519
+Jews
+
+4.2
+00:00:23,519 --> 00:00:23,899
+can't
+
+4.3
+00:00:23,899 --> 00:00:24,059
+be
+
+4.4
+00:00:24,059 --> 00:00:24,199
+in
+
+4.5
+00:00:24,199 --> 00:00:24,660
+power.
+
+4.6
+00:00:25,059 --> 00:00:25,739
+People
+
+4.7
+00:00:25,739 --> 00:00:25,980
+go,
+
+4.8
+00:00:26,120 --> 00:00:26,280
+whoa,
+
+4.9
+00:00:26,440 --> 00:00:26,500
+I
+
+4.10
+00:00:26,500 --> 00:00:26,660
+don't
+
+4.11
+00:00:26,660 --> 00:00:26,820
+hear
+
+4.12
+00:00:26,820 --> 00:00:27,079
+people
+
+4.13
+00:00:27,079 --> 00:00:27,399
+talk
+
+4.14
+00:00:27,399 --> 00:00:27,640
+like
+
+4.15
+00:00:27,640 --> 00:00:27,960
+that.
+
+4.16
+00:00:28,320 --> 00:00:28,839
+What
+
+4.17
+00:00:28,839 --> 00:00:28,920
+do
+
+4.18
+00:00:28,920 --> 00:00:29,000
+you
+
+4.19
+00:00:29,000 --> 00:00:29,199
+mean?
+

endpoints/__init__.py

@@ -4,3 +4,4 @@ from .subway import router as subway_router
 from .minecraft import router as minecraft_router
 from .basic import router as basic_router
 from .video_editor_cut import router as video_editor
+from .rumble import router as rumble_router

endpoints/rumble.py

@@ -0,0 +1,85 @@
+from fastapi import Body, HTTPException
+from fastapi.responses import StreamingResponse
+from typing import Optional
+from fastapi import APIRouter
+from generate_rumble_clips import generate_rumble_clips
+from endpoints.utils import remove_content_from_dir
+from endpoints.utils import rumble_link_scraper
+import io
+import zipfile
+import pydantic
+import os
+
+router = APIRouter()
+
+class ClipData(pydantic.BaseModel):
+	start: str
+	end: str
+	name: str
+	clip_link: Optional[str] = None
+
+class RumbleData(pydantic.BaseModel):
+	video_url: str
+	color: str
+	size: int
+	font: str
+	credit: str
+	credit_size: int
+	clips: list[ClipData]
+
+
+@router.post("/rumble")
+async def create_rumble_video(request_data: RumbleData = Body(...)):
+
+	# Extract values from the request_data dictionary
+	video_url = request_data.video_url
+	color = request_data.color
+	size = request_data.size
+	font = request_data.font
+	credit = request_data.credit
+	credit_size = request_data.credit_size
+	clips = request_data.clips
+
+	options = {
+		"video_url": video_url,
+		"clips": clips,
+		"font_color": color,
+		"font_size": size,
+		"font_family": font,
+		"border_size": 1,
+		"credit": credit,
+		"credit_size": credit_size,
+	}
+
+	try:
+		# scrape the web for the rumble link
+		options["video_url"] = None
+
+		print(options["video_url"])
+
+		# Create the directory if it doesn't exist
+		if not os.path.exists("./data/rumble/clips"):
+			os.makedirs("./data/rumble/clips")
+
+		# Generate the clips
+		clips = generate_rumble_clips(options)
+		
+		# Create a zip archive in memory
+		zip_buffer = io.BytesIO()
+		with zipfile.ZipFile(zip_buffer, "w") as zip_file:
+			for clip in clips:
+				zip_file.write(clip.clip_link, arcname=f"clips/{clip.name}.mp4")
+
+		# Reset the buffer position to the beginning
+		zip_buffer.seek(0)		
+
+		# remove the clips directory content
+		remove_content_from_dir("./data/rumble/clips")
+
+		# Return the zip archive as a StreamingResponse
+		return StreamingResponse(io.BytesIO(zip_buffer.read()), media_type="application/zip", headers={
+			'Content-Disposition': 'attachment; filename="clips.zip"'
+		})
+	except Exception as e:
+		# Handle exceptions appropriately
+		raise HTTPException(status_code=500, detail=str(e))

endpoints/utils.py

@@ -1,10 +1,14 @@
 import os
 import shutil
 import time
+import httpx
+from bs4 import BeautifulSoup
+
 
 def get_file_name(prefix: str) -> str:
 	return f"{prefix}_{time.strftime('%Y%m%d-%H%M%S')}.mp4"
 
+
 def remove_content_from_dir(folder):
 	for filename in os.listdir(folder):
 		file_path = os.path.join(folder, filename)
@@ -14,4 +18,40 @@ def remove_content_from_dir(folder):
 			elif os.path.isdir(file_path):
 				shutil.rmtree(file_path)
 		except Exception as e:
-			print('Failed to delete %s. Reason: %s' % (file_path, e))
+			print('Failed to delete %s. Reason: %s' % (file_path, e))
+
+
+async def rumble_link_scraper(rumble_link):
+	url = "https://imgpanda.com/wp-admin/admin-ajax.php"
+	downloader_type = "Rumble"
+	server_name = "Server2"
+
+	headers = {
+		"Content-Type": "application/x-www-form-urlencoded",
+	}
+
+	data = {
+		"action": "ajax_call_custom",
+		"SocialDownUrl": rumble_link,
+		"downloaderType": downloader_type,
+		"serverName": server_name,
+	}
+
+	async with httpx.AsyncClient() as client:
+		response = await client.post(url, headers=headers, data=data)
+
+	if response.status_code == 200:
+		soup = BeautifulSoup(response.text, 'html.parser')
+		inner_html_string = ' '.join(response.text.split(' ')[1:-1])
+		inner_soup = BeautifulSoup(inner_html_string, 'html.parser')
+
+		# all a-tags within the class "social-download-result"
+		a_tags = inner_soup.select('a')
+		last_a_tag = a_tags[-1]
+
+		if last_a_tag:
+			return last_a_tag.get('href').replace('\\', '').replace('"', '')
+		else:
+			print('No matching <a> tag found within the class "social-download-result"')
+	else:
+		print(f"Error: {response.status_code}")

generate_rumble_clips.py

@@ -0,0 +1,101 @@
+import os
+from utils.create_subtitles import hex_to_ffmpeg_color, create_subtitles
+from utils.transcribe_audio import transcribe_audio
+from utils.video_to_audio import extract_audio
+import subprocess
+
+
+def generate_rumble_clips(customization_options, output_directory="./data/rumble/clips"):
+	customization_options['font_color'] = hex_to_ffmpeg_color(customization_options['font_color'])
+	audio_file_path = "./data/rumble/audio.mp3"
+	subtitle_file = "./data/rumble/subtitles.srt"
+
+	clips = []
+
+	for idx, clip in enumerate(customization_options['clips']):
+		# Generate the clip
+		video_clip_path = cut_video(customization_options['video_url'], clip.start, clip.end, output_directory, idx)
+		clip.clip_link = video_clip_path
+		clips.append(clip)
+
+		# Generate the subtitles
+		extract_audio(video_clip_path, audio_file_path)
+		transcription = transcribe_audio(audio_file_path)
+		create_subtitles(transcription, subtitle_file)
+
+		# Add the subtitle to the video
+		subtitle_video_clip = add_subtitle(video_clip_path, subtitle_file, customization_options)
+
+		# Add the text to the video
+		video_with_text = add_text_to_video(subtitle_video_clip, customization_options)
+
+		video_sped_up = speed_up_video(video_with_text)
+		clip.clip_link = video_sped_up
+	
+	return clips
+
+
+def cut_video(input_url, start_time, end_time, output_directory, idx):
+	output_filepath = os.path.join(output_directory, f'output_clip_{idx}.mp4')
+
+	command = [
+		'ffmpeg',
+		'-i', input_url,
+		'-ss', start_time,
+		'-to', end_time,
+		'-y', output_filepath
+	]
+
+	subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+
+	return output_filepath
+
+
+def add_subtitle(video_path, subtitle_file, customization_options):
+	output_filepath = video_path.replace('.mp4', '_with_subtitle.mp4')
+
+	subtitle_cmd = [
+		"ffmpeg",
+		"-i", video_path,
+		"-vf", f"subtitles={subtitle_file}:force_style='Fontsize={customization_options['font_size']},PrimaryColour={customization_options['font_color']},Fontname={customization_options['font_family']},MarginV=50'",
+		"-c:a", "copy",
+		'-y', output_filepath
+	]
+
+	subprocess.run(subtitle_cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+
+	return output_filepath
+
+
+def add_text_to_video(video_path, customization_options):
+	output_filepath = video_path.replace('.mp4', '_with_text.mp4')
+
+	text_cmd = [
+		"ffmpeg",
+		"-i", video_path,
+		'-vf', f'drawtext=text=\'{customization_options["credit"]}\':fontcolor=white:fontsize={customization_options["credit_size"]}:box=1:[email protected]:boxborderw=5:x=(w-text_w)/2:y=h-text_h-10',
+		"-c:a", "copy",
+		'-y', output_filepath
+	]
+
+	subprocess.run(text_cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+
+	return output_filepath
+
+
+def speed_up_video(video_path):
+    output_filepath = video_path.replace('.mp4', '_sped_up.mp4')
+
+    speed_up_cmd = [
+        "ffmpeg",
+        "-i", video_path,
+        "-filter:v", "setpts=0.66*PTS",  # Speed up by 1.5x (1/1.5 = 0.6666)
+        "-filter:a", "atempo=1.5",  # Speed up audio as well
+        "-c:v", "libx264", "-c:a", "aac",
+        "-strict", "experimental",
+        "-y", output_filepath
+    ]
+
+    subprocess.run(speed_up_cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+
+    return output_filepath

server.py

@@ -3,7 +3,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from fastapi.staticfiles import StaticFiles
 from fastapi.responses import FileResponse
 import uvicorn
-from endpoints import subway_router, minecraft_router, basic_router, video_editor
+from endpoints import subway_router, minecraft_router, basic_router, video_editor, rumble_router
 
 app = FastAPI()
 
@@ -29,6 +29,7 @@ app.include_router(subway_router, prefix="/generate-subtitles")
 app.include_router(minecraft_router, prefix="/generate-subtitles")
 app.include_router(basic_router, prefix="/generate-subtitles")
 app.include_router(video_editor, prefix="/video-editor")
+app.include_router(rumble_router, prefix="/generate-subtitles")
 
 if __name__ == "__main__":
 	uvicorn.run("server:app", host="localhost", port=8000, reload=True)

static/index.html

@@ -11,15 +11,40 @@
 		<h1>BrainRotTok</h1>
 		<div class="form-buttons">
 			<ul>
-				<li><button class="show-form" data-form-id="basic-form">BASIC</button></li>
-				<li><button class="show-form" data-form-id="subway-form">SUBWAY</button></li>
 				<li>
-					<button class="show-form" data-form-id="minecraft-form">MINECRAFT</button>
+					<button class="show-form" data-form-id="basic-form">
+						BASIC
+					</button>
 				</li>
 				<li>
-					<button class="show-form" data-form-id="video-cutter-form" disabled>VIDEO CUTTER</button>
+					<button class="show-form" data-form-id="subway-form">
+						SUBWAY
+					</button>
 				</li>
-			</ul>	
+				<li>
+					<button class="show-form" data-form-id="minecraft-form">
+						MINECRAFT
+					</button>
+				</li>
+				<li>
+					<button
+						class="show-form"
+						data-form-id="video-cutter-form"
+						disabled
+					>
+						VIDEO CUTTER
+					</button>
+				</li>
+				<li>
+					<button
+						class="show-form"
+						data-form-id="rumble-form"
+						disabled
+					>
+						RUMBLE
+					</button>
+				</li>
+			</ul>
 		</div>
 		<div id="forms-container">
 			<form
@@ -53,7 +78,11 @@
 				enctype="multipart/form-data"
 			>
 				<h2 class="form-title">SUBWAY</h2>
-				<p>Merge a top and bottom video together with subtitles. The top video will have no sound, while the bottom video will have its subtitles generated.</p>
+				<p>
+					Merge a top and bottom video together with subtitles. The
+					top video will have no sound, while the bottom video will
+					have its subtitles generated.
+				</p>
 				<label for="top_video">Top Video:</label>
 				<input type="file" id="top_video" name="top_video" />
 				<label for="bottom_video">Bottom Video:</label>
@@ -78,7 +107,10 @@
 				enctype="multipart/form-data"
 			>
 				<h2 class="form-title">MINECRAFT</h2>
-				<p> A background video playing while a script is being read by an AI voice with subtitles.</p>
+				<p>
+					A background video playing while a script is being read by
+					an AI voice with subtitles.
+				</p>
 				<label for="video">Video:</label>
 				<input type="file" id="video" name="video" />
 				<label for="subtitles">Subtitles:</label>
@@ -95,6 +127,29 @@
 				<input type="number" id="credit_size" name="credit_size" />
 				<button type="submit">Submit</button>
 			</form>
+			<form
+				id="rumble-form"
+				class="hidden"
+				action="/generate-subtitles/rumble"
+				method="post"
+				enctype="multipart/form-data"
+			>
+				<h2 class="form-title">RUMBLE</h2>
+				<p>Generates subtitles for Rumble videos.</p>
+				<label for="video_url">Video Url:</label>
+				<input type="text" id="video_url" name="video_url" />
+				<label for="color">Color:</label>
+				<input type="text" id="color" name="color" />
+				<label for="size">Size:</label>
+				<input type="number" id="size" name="size" />
+				<label for="font">Font:</label>
+				<input type="text" id="font" name="font" />
+				<label for="credit">Credit:</label>
+				<input type="text" id="credit" name="credit" />
+				<label for="credit_size">Credit Size:</label>
+				<input type="number" id="credit_size" name="credit_size" />
+				<button type="submit">Submit</button>
+			</form>
 			<form
 				id="video-cutter-form"
 				class="hidden"