mirror of
https://github.com/harry0703/MoneyPrinterTurbo.git
synced 2026-02-21 16:37:21 +08:00
wip
This commit is contained in:
faycal
parent
8619e3dd2f
commit
67da6765ad
138
function/main.py
138
function/main.py
@ -2,9 +2,14 @@ import os
|
||||
import requests
|
||||
import json
|
||||
import time
|
||||
import logging
|
||||
from tiktokautouploader import upload_tiktok
|
||||
import functions_framework
|
||||
|
||||
# Configure logging
|
||||
logging.basicConfig(level=logging.INFO)
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
api_base_url = os.getenv("API_URL")
|
||||
headers = {"Content-Type": "application/json"}
|
||||
|
||||
@ -31,78 +36,105 @@ bgm_volume = 0.2
|
||||
|
||||
@functions_framework.http
|
||||
def main(request=None):
|
||||
|
||||
generate_script_url = f"{api_base_url}/scripts"
|
||||
script_payload = {
|
||||
"video_subject": video_subject,
|
||||
"video_language": video_language,
|
||||
"paragraph_number": 1
|
||||
}
|
||||
response = requests.post(generate_script_url, headers=headers, data=json.dumps(script_payload))
|
||||
if response.status_code != 200:
|
||||
return f"Error generating script: {response.status_code}", 500
|
||||
upload_music_url = f"{api_base_url}/musics"
|
||||
files = {'file': open(song_file_path, 'rb')}
|
||||
try:
|
||||
response = requests.post(upload_music_url, files=files)
|
||||
response.raise_for_status()
|
||||
bgm_file = response.json()['data']['file']
|
||||
except requests.exceptions.RequestException as e:
|
||||
logger.error(f"Error uploading music: {e}")
|
||||
return f"Error uploading music: {e}", 500
|
||||
except json.JSONDecodeError as e:
|
||||
logger.error(f"Error decoding JSON response from music upload: {e}")
|
||||
return "Error decoding JSON response from music upload", 500
|
||||
|
||||
try:
|
||||
generate_script_url = f"{api_base_url}/scripts"
|
||||
script_payload = {
|
||||
"video_subject": video_subject,
|
||||
"video_language": video_language,
|
||||
"paragraph_number": 1
|
||||
}
|
||||
response = requests.post(generate_script_url, headers=headers, data=json.dumps(script_payload))
|
||||
response.raise_for_status()
|
||||
video_script = response.json()['data']['video_script']
|
||||
except json.JSONDecodeError:
|
||||
except requests.exceptions.RequestException as e:
|
||||
logger.error(f"Error generating script: {e}")
|
||||
return f"Error generating script: {e}", 500
|
||||
except json.JSONDecodeError as e:
|
||||
logger.error(f"Error decoding JSON response from script generation: {e}")
|
||||
return "Error decoding JSON response from script generation", 500
|
||||
|
||||
generate_keywords_url = f"{api_base_url}/terms"
|
||||
keywords_payload = {
|
||||
"video_subject": video_subject,
|
||||
"video_script": video_script,
|
||||
"amount": 5
|
||||
}
|
||||
response = requests.post(generate_keywords_url, headers=headers, data=json.dumps(keywords_payload))
|
||||
if response.status_code != 200:
|
||||
return f"Error generating keywords: {response.status_code}", 500
|
||||
try:
|
||||
generate_keywords_url = f"{api_base_url}/terms"
|
||||
keywords_payload = {
|
||||
"video_subject": video_subject,
|
||||
"video_script": video_script,
|
||||
"amount": 5
|
||||
}
|
||||
response = requests.post(generate_keywords_url, headers=headers, data=json.dumps(keywords_payload))
|
||||
response.raise_for_status()
|
||||
video_terms = response.json()['data']['video_terms']
|
||||
except json.JSONDecodeError:
|
||||
except requests.exceptions.RequestException as e:
|
||||
logger.error(f"Error generating keywords: {e}")
|
||||
return f"Error generating keywords: {e}", 500
|
||||
except json.JSONDecodeError as e:
|
||||
logger.error(f"Error decoding JSON response from keywords generation: {e}")
|
||||
return "Error decoding JSON response from keywords generation", 500
|
||||
|
||||
generate_video_url = f"{api_base_url}/videos"
|
||||
video_payload = {
|
||||
"video_subject": video_subject,
|
||||
"video_script": video_script,
|
||||
"video_terms": video_terms,
|
||||
"video_aspect": video_aspect_ratio,
|
||||
"video_concat_mode": video_concat_mode,
|
||||
"video_transition_mode": video_transition_mode,
|
||||
"video_clip_duration": video_clip_duration,
|
||||
"video_count": video_count,
|
||||
"video_source": video_source,
|
||||
"video_language": video_language,
|
||||
"voice_name": speech_voice,
|
||||
"voice_volume": speech_volume,
|
||||
"voice_rate": speech_rate,
|
||||
"bgm_type": "custom",
|
||||
"bgm_file": bgm_file,
|
||||
"bgm_volume": bgm_volume,
|
||||
"subtitle_enabled": False,
|
||||
}
|
||||
|
||||
response = requests.post(generate_video_url, headers=headers, data=json.dumps(video_payload))
|
||||
if response.status_code != 200:
|
||||
return f"Error generating video: {response.status_code}", 500
|
||||
try:
|
||||
generate_video_url = f"{api_base_url}/videos"
|
||||
video_payload = {
|
||||
"video_subject": video_subject,
|
||||
"video_script": video_script,
|
||||
"video_terms": video_terms,
|
||||
"video_aspect": video_aspect_ratio,
|
||||
"video_concat_mode": video_concat_mode,
|
||||
"video_transition_mode": video_transition_mode,
|
||||
"video_clip_duration": video_clip_duration,
|
||||
"video_count": video_count,
|
||||
"video_source": video_source,
|
||||
"video_language": video_language,
|
||||
"voice_name": speech_voice,
|
||||
"voice_volume": speech_volume,
|
||||
"voice_rate": speech_rate,
|
||||
"bgm_type": "custom",
|
||||
"bgm_file": bgm_file,
|
||||
"bgm_volume": bgm_volume,
|
||||
"subtitle_enabled": False,
|
||||
}
|
||||
response = requests.post(generate_video_url, headers=headers, data=json.dumps(video_payload))
|
||||
response.raise_for_status()
|
||||
task_id = response.json()['data']['task_id']
|
||||
except json.JSONDecodeError:
|
||||
except requests.exceptions.RequestException as e:
|
||||
logger.error(f"Error generating video: {e}")
|
||||
return f"Error generating video: {e}", 500
|
||||
except json.JSONDecodeError as e:
|
||||
logger.error(f"Error decoding JSON response from video generation: {e}")
|
||||
return "Error decoding JSON response from video generation", 500
|
||||
|
||||
time.sleep(float(os.getenv("TIME_BEFORE_DOWNLOAD")))
|
||||
|
||||
download_url = f"{api_base_url}/download/{task_id}/final-1.mp4"
|
||||
response = requests.get(download_url, headers=headers)
|
||||
if response.status_code != 200:
|
||||
return f"Error downloading video: {response.status_code}", 500
|
||||
with open('output_video.mp4', 'wb') as f:
|
||||
f.write(response.content)
|
||||
try:
|
||||
download_url = f"{api_base_url}/download/{task_id}/final-1.mp4"
|
||||
response = requests.get(download_url, headers=headers)
|
||||
response.raise_for_status()
|
||||
with open('output_video.mp4', 'wb') as f:
|
||||
f.write(response.content)
|
||||
except requests.exceptions.RequestException as e:
|
||||
logger.error(f"Error downloading video: {e}")
|
||||
return f"Error downloading video: {e}", 500
|
||||
|
||||
video_path = "output_video.mp4"
|
||||
description = os.getenv("VIDEO_DESCRIPTION")
|
||||
account_name = os.getenv("ACCOUNT_NAME")
|
||||
hashtags = ['fyp']
|
||||
|
||||
upload_tiktok(video=video_path, description=description, accountname=account_name)
|
||||
try:
|
||||
upload_tiktok(video=video_path, description=description, accountname=account_name)
|
||||
except Exception as e:
|
||||
logger.error(f"Error uploading to TikTok: {e}")
|
||||
return f"Error uploading to TikTok: {e}", 500
|
||||
|
||||
return "Function executed successfully"
|
||||
Loading…
Reference in New Issue
Block a user