Drugsosos
2 years ago
commit
cd24608f86
@ -1,29 +1,86 @@
|
|||||||
REDDIT_CLIENT_ID=""
|
|
||||||
REDDIT_CLIENT_SECRET=""
|
|
||||||
|
|
||||||
REDDIT_USERNAME=""
|
REDDIT_CLIENT_ID="" #fFAGRNJru1FTz70BzhT3Zg
|
||||||
REDDIT_PASSWORD=""
|
#EXPLANATION the ID of your Reddit app of SCRIPT type
|
||||||
|
#RANGE 12:30
|
||||||
|
#MATCH_REGEX [-a-zA-Z0-9._~+/]+=*$
|
||||||
|
#OOB_ERROR The ID should be over 12 and under 30 characters, double check your input.
|
||||||
|
|
||||||
# Valid options are "yes" and "no"
|
REDDIT_CLIENT_SECRET="" #fFAGRNJru1FTz70BzhT3Zg
|
||||||
REDDIT_2FA=""
|
#EXPLANATION the SECRET of your Reddit app of SCRIPT type
|
||||||
|
#RANGE 20:40
|
||||||
|
#MATCH_REGEX [-a-zA-Z0-9._~+/]+=*$
|
||||||
|
#OOB_ERROR The secret should be over 20 and under 40 characters, double check your input.
|
||||||
|
|
||||||
#If no, it will ask you a thread link to extract the thread, if yes it will randomize it.
|
REDDIT_USERNAME="" #asdfghjkl
|
||||||
RANDOM_THREAD="yes"
|
#EXPLANATION the username of your reddit account
|
||||||
|
#RANGE 3:20
|
||||||
|
#MATCH_REGEX [-_0-9a-zA-Z]+$
|
||||||
|
#OOB_ERROR A username HAS to be between 3 and 20 characters
|
||||||
|
|
||||||
# Valid options are "light" and "dark"
|
REDDIT_PASSWORD="" #fFAGRNJru1FTz70BzhT3Zg
|
||||||
THEME=""
|
#EXPLANATION the password of your reddit account
|
||||||
|
#RANGE 8:None
|
||||||
|
#OOB_ERROR Password too short
|
||||||
|
|
||||||
# Enter a subreddit, e.g. "AskReddit"
|
#OPTIONAL
|
||||||
SUBREDDIT=""
|
RANDOM_THREAD="no"
|
||||||
|
# If set to no, it will ask you a thread link to extract the thread, if yes it will randomize it. Default: "no"
|
||||||
|
|
||||||
# Filters the comments by range of lenght (min and max characters)
|
REDDIT_2FA="" #no
|
||||||
# Min has to be less or equal to max
|
#MATCH_REGEX ^(yes|no)
|
||||||
# DO NOT INSERT ANY SPACES BETWEEN THE COMMA AND THE VALUES
|
#EXPLANATION Whether you have Reddit 2FA enabled, Valid options are "yes" and "no"
|
||||||
COMMENT_LENGTH_RANGE = "min,max"
|
|
||||||
|
|
||||||
# Range is 0 -> 1
|
SUBREDDIT="AskReddit"
|
||||||
OPACITY="0.9"
|
#EXPLANATION what subreddit to pull posts from, the name of the sub, not the URL
|
||||||
|
#RANGE 3:20
|
||||||
|
#MATCH_REGEX [_0-9a-zA-Z]+$
|
||||||
|
#OOB_ERROR A subreddit name HAS to be between 3 and 20 characters
|
||||||
|
|
||||||
# The absolute path of the folder where you want to save the final video
|
ALLOW_NSFW="False"
|
||||||
# If empty or wrong, the path will be 'assets/'
|
#EXPLANATION Whether to allow NSFW content, True or False
|
||||||
FINAL_VIDEO_PATH=""
|
#MATCH_REGEX ^(True|False)$
|
||||||
|
|
||||||
|
POST_ID=""
|
||||||
|
#MATCH_REGEX ^((?!://|://).)*$
|
||||||
|
#EXPLANATION Used if you want to use a specific post. example of one is urdtfx
|
||||||
|
|
||||||
|
THEME="LIGHT" #dark
|
||||||
|
#EXPLANATION sets the Reddit theme, either LIGHT or DARK
|
||||||
|
#MATCH_REGEX ^(dark|light|DARK|LIGHT)$
|
||||||
|
|
||||||
|
TIMES_TO_RUN="" #2
|
||||||
|
#EXPLANATION used if you want to run multiple times. set to an int e.g. 4 or 29 and leave blank for 1
|
||||||
|
|
||||||
|
MAX_COMMENT_LENGTH="500" #500
|
||||||
|
#EXPLANATION max number of characters a comment can have. default is 500
|
||||||
|
#RANGE 0:10000
|
||||||
|
#MATCH_TYPE int
|
||||||
|
#OOB_ERROR the max comment length should be between 0 and 10000
|
||||||
|
|
||||||
|
OPACITY="1" #.8
|
||||||
|
#EXPLANATION Sets the opacity of the comments when overlayed over the background
|
||||||
|
#RANGE 0:1
|
||||||
|
#MATCH_TYPE float
|
||||||
|
#OOB_ERROR The opacity HAS to be between 0 and 1
|
||||||
|
|
||||||
|
# If you want to translate the comments to another language, set the language code here.
|
||||||
|
# If empty, no translation will be done.
|
||||||
|
POSTLANG=""
|
||||||
|
#EXPLANATION Activates the translation feature, set the language code for translate or leave blank
|
||||||
|
|
||||||
|
TTSCHOICE="Polly"
|
||||||
|
#EXPLANATION the backend used for TTS. Without anything specified, the user will be prompted to choose one.
|
||||||
|
# IMPORTANT NOTE: if you use translate, you need to set this to googletranslate or tiktok and use custom voice in your language
|
||||||
|
|
||||||
|
STREAMLABS_VOICE="Joanna"
|
||||||
|
#EXPLANATION Sets the voice for the Streamlabs Polly TTS Engine. Check the file for more information on different voices.
|
||||||
|
|
||||||
|
AWS_VOICE="Joanna"
|
||||||
|
#EXPLANATION Sets the voice for the AWS Polly TTS Engine. Check the file for more information on different voices.
|
||||||
|
|
||||||
|
TIKTOK_VOICE="en_us_006"
|
||||||
|
#EXPLANATION Sets the voice for the TikTok TTS Engine. Check the file for more information on different voices.
|
||||||
|
|
||||||
|
#OPTIONAL
|
||||||
|
STORYMODE="False"
|
||||||
|
# IN-PROGRESS - not yet implemented
|
||||||
|
|||||||
@ -0,0 +1 @@
|
|||||||
|
* text=auto eol=lf
|
||||||
@ -0,0 +1,28 @@
|
|||||||
|
---
|
||||||
|
name: Bug report
|
||||||
|
about: Create a report to help us improve
|
||||||
|
title: ""
|
||||||
|
labels: bug
|
||||||
|
assignees: ""
|
||||||
|
---
|
||||||
|
|
||||||
|
**Describe the bug**
|
||||||
|
A clear and concise description of what the bug is.
|
||||||
|
|
||||||
|
**To Reproduce**
|
||||||
|
Steps to reproduce the behavior:
|
||||||
|
|
||||||
|
**Expected behavior**
|
||||||
|
A clear and concise description of what you expected to happen.
|
||||||
|
|
||||||
|
**Screenshots**
|
||||||
|
If applicable, add screenshots to help explain your problem.
|
||||||
|
|
||||||
|
**System (please complete the following information):**
|
||||||
|
|
||||||
|
- Python Version: [e.g. Python 3.6]
|
||||||
|
- OS: [e.g. Windows 11]
|
||||||
|
- App version / Branch [e.g. latest, V2.0, master, develop ]
|
||||||
|
|
||||||
|
**Additional context**
|
||||||
|
Add any other context about the problem here.
|
||||||
@ -0,0 +1,20 @@
|
|||||||
|
---
|
||||||
|
name: Feature request
|
||||||
|
about: Suggest an idea for this project
|
||||||
|
title: ''
|
||||||
|
labels: enhancement
|
||||||
|
assignees: ''
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
**Is your feature request related to a problem? Please describe.**
|
||||||
|
A clear and concise description of what the problem is. Ex. I'm always frustrated when [...]
|
||||||
|
|
||||||
|
**Describe the solution you'd like**
|
||||||
|
A clear and concise description of what you want to happen.
|
||||||
|
|
||||||
|
**Describe alternatives you've considered** (optional)
|
||||||
|
A clear and concise description of any alternative solutions or features you've considered.
|
||||||
|
|
||||||
|
**Additional context** (optional)
|
||||||
|
Add any other context or screenshots about the feature request here.
|
||||||
@ -0,0 +1,25 @@
|
|||||||
|
# Description
|
||||||
|
|
||||||
|
<!-- Please include a summary of the change and which issue is fixed. Please also include relevant context. List any dependencies that are required for this change. -->
|
||||||
|
|
||||||
|
# Issue Fixes
|
||||||
|
|
||||||
|
<!-- Fixes #(issue) if relevant-->
|
||||||
|
|
||||||
|
None
|
||||||
|
|
||||||
|
# Checklist:
|
||||||
|
|
||||||
|
- [ ] I am pushing changes to the **develop** branch
|
||||||
|
- [ ] I am using the recommended development environment
|
||||||
|
- [ ] I have performed a self-review of my own code
|
||||||
|
- [ ] I have commented my code, particularly in hard-to-understand areas
|
||||||
|
- [ ] I have formatted and linted my code using python-black and pylint
|
||||||
|
- [ ] I have cleaned up unnecessary files
|
||||||
|
- [ ] My changes generate no new warnings
|
||||||
|
- [ ] My changes follow the existing code-style
|
||||||
|
- [ ] My changes are relevant to the project
|
||||||
|
|
||||||
|
# Any other information (e.g how to test the changes)
|
||||||
|
|
||||||
|
None
|
||||||
@ -0,0 +1,11 @@
|
|||||||
|
# To get started with Dependabot version updates, you'll need to specify which
|
||||||
|
# package ecosystems to update and where the package manifests are located.
|
||||||
|
# Please see the documentation for all configuration options:
|
||||||
|
# https://docs.github.com/github/administering-a-repository/configuration-options-for-dependency-updates
|
||||||
|
|
||||||
|
version: 2
|
||||||
|
updates:
|
||||||
|
- package-ecosystem: "pip" # See documentation for possible values
|
||||||
|
directory: "/" # Location of package manifests
|
||||||
|
schedule:
|
||||||
|
interval: "daily"
|
||||||
@ -0,0 +1,40 @@
|
|||||||
|
name: 'Stale issue handler'
|
||||||
|
on:
|
||||||
|
workflow_dispatch:
|
||||||
|
schedule:
|
||||||
|
- cron: '0 0 * * *'
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
stale:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
- uses: actions/stale@main
|
||||||
|
id: stale-issue
|
||||||
|
name: stale-issue
|
||||||
|
with:
|
||||||
|
stale-issue-message: 'This issue is stale because it has been open 7 days with no activity. Remove stale label or comment, or this will be closed in 10 days.'
|
||||||
|
close-issue-message: 'Issue closed due to being stale. Please reopen if issue persists in latest version.'
|
||||||
|
days-before-stale: 7
|
||||||
|
days-before-close: 10
|
||||||
|
stale-issue-label: 'stale'
|
||||||
|
close-issue-label: 'outdated'
|
||||||
|
exempt-issue-labels: 'enhancement,keep,blocked'
|
||||||
|
exempt-all-issue-milestones: true
|
||||||
|
operations-per-run: 300
|
||||||
|
remove-stale-when-updated: true
|
||||||
|
|
||||||
|
- uses: actions/stale@main
|
||||||
|
id: stale-pr
|
||||||
|
name: stale-pr
|
||||||
|
with:
|
||||||
|
stale-pr-message: 'This pull request is stale as it has been open for 7 days with no activity. Remove stale label or comment, or this will be closed in 10 days.'
|
||||||
|
close-pr-message: 'Pull request closed due to being stale.'
|
||||||
|
days-before-stale: 7
|
||||||
|
days-before-close: 10
|
||||||
|
close-pr-label: 'outdated'
|
||||||
|
stale-pr-label: 'stale'
|
||||||
|
exempt-pr-labels: 'keep,blocked,before next release,after next release'
|
||||||
|
exempt-all-pr-milestones: true
|
||||||
|
operations-per-run: 300
|
||||||
|
remove-stale-when-updated: true
|
||||||
|
|
||||||
@ -0,0 +1,127 @@
|
|||||||
|
# Contributor Covenant Code of Conduct
|
||||||
|
|
||||||
|
## Our Pledge
|
||||||
|
|
||||||
|
We as members, contributors, and leaders pledge to make participation in our
|
||||||
|
community a harassment-free experience for everyone, regardless of age, body
|
||||||
|
size, visible or invisible disability, ethnicity, sex characteristics, gender
|
||||||
|
identity and expression, level of experience, education, socio-economic status,
|
||||||
|
nationality, personal appearance, race, religion, or sexual identity
|
||||||
|
and orientation.
|
||||||
|
|
||||||
|
We pledge to act and interact in ways that contribute to an open, welcoming,
|
||||||
|
diverse, inclusive, and healthy community.
|
||||||
|
|
||||||
|
## Our Standards
|
||||||
|
|
||||||
|
Examples of behavior that contributes to a positive environment for our
|
||||||
|
community include:
|
||||||
|
|
||||||
|
- Demonstrating empathy and kindness toward other people
|
||||||
|
- Being respectful of differing opinions, viewpoints, and experiences
|
||||||
|
- Giving and gracefully accepting constructive feedback
|
||||||
|
- Accepting responsibility and apologizing to those affected by our mistakes,
|
||||||
|
and learning from the experience
|
||||||
|
- Focusing on what is best not just for us as individuals, but for the
|
||||||
|
overall community
|
||||||
|
|
||||||
|
Examples of unacceptable behavior include:
|
||||||
|
|
||||||
|
- The use of sexualized language or imagery, and sexual attention or
|
||||||
|
advances of any kind
|
||||||
|
- Trolling, insulting or derogatory comments, and personal or political attacks
|
||||||
|
- Public or private harassment
|
||||||
|
- Publishing others' private information, such as a physical or email
|
||||||
|
address, without their explicit permission
|
||||||
|
- Other conduct which could reasonably be considered inappropriate in a
|
||||||
|
professional setting
|
||||||
|
|
||||||
|
## Enforcement Responsibilities
|
||||||
|
|
||||||
|
Community leaders are responsible for clarifying and enforcing our standards of
|
||||||
|
acceptable behavior and will take appropriate and fair corrective action in
|
||||||
|
response to any behavior that they deem inappropriate, threatening, offensive,
|
||||||
|
or harmful.
|
||||||
|
|
||||||
|
Community leaders have the right and responsibility to remove, edit, or reject
|
||||||
|
comments, commits, code, wiki edits, issues, and other contributions that are
|
||||||
|
not aligned to this Code of Conduct, and will communicate reasons for moderation
|
||||||
|
decisions when appropriate.
|
||||||
|
|
||||||
|
## Scope
|
||||||
|
|
||||||
|
This Code of Conduct applies within all community spaces, and also applies when
|
||||||
|
an individual is officially representing the community in public spaces.
|
||||||
|
Examples of representing our community include using an official e-mail address,
|
||||||
|
posting via an official social media account, or acting as an appointed
|
||||||
|
representative at an online or offline event.
|
||||||
|
|
||||||
|
## Enforcement
|
||||||
|
|
||||||
|
Instances of abusive, harassing, or otherwise unacceptable behavior may be
|
||||||
|
reported to the community leaders responsible for enforcement at the [discord server](https://discord.gg/yqNvvDMYpq).
|
||||||
|
All complaints will be reviewed and investigated promptly and fairly.
|
||||||
|
|
||||||
|
All community leaders are obligated to respect the privacy and security of the
|
||||||
|
reporter of any incident.
|
||||||
|
|
||||||
|
## Enforcement Guidelines
|
||||||
|
|
||||||
|
Community leaders will follow these Community Impact Guidelines in determining
|
||||||
|
the consequences for any action they deem in violation of this Code of Conduct:
|
||||||
|
|
||||||
|
### 1. Correction
|
||||||
|
|
||||||
|
**Community Impact**: Use of inappropriate language or other behavior deemed
|
||||||
|
unprofessional or unwelcome in the community.
|
||||||
|
|
||||||
|
**Consequence**: A private, written warning from community leaders, providing
|
||||||
|
clarity around the nature of the violation and an explanation of why the
|
||||||
|
behavior was inappropriate. A public apology may be requested.
|
||||||
|
|
||||||
|
### 2. Warning
|
||||||
|
|
||||||
|
**Community Impact**: A violation through a single incident or series
|
||||||
|
of actions.
|
||||||
|
|
||||||
|
**Consequence**: A warning with consequences for continued behavior. No
|
||||||
|
interaction with the people involved, including unsolicited interaction with
|
||||||
|
those enforcing the Code of Conduct, for a specified period of time. This
|
||||||
|
includes avoiding interactions in community spaces as well as external channels
|
||||||
|
like social media. Violating these terms may lead to a temporary or
|
||||||
|
permanent ban.
|
||||||
|
|
||||||
|
### 3. Temporary Ban
|
||||||
|
|
||||||
|
**Community Impact**: A serious violation of community standards, including
|
||||||
|
sustained inappropriate behavior.
|
||||||
|
|
||||||
|
**Consequence**: A temporary ban from any sort of interaction or public
|
||||||
|
communication with the community for a specified period of time. No public or
|
||||||
|
private interaction with the people involved, including unsolicited interaction
|
||||||
|
with those enforcing the Code of Conduct, is allowed during this period.
|
||||||
|
Violating these terms may lead to a permanent ban.
|
||||||
|
|
||||||
|
### 4. Permanent Ban
|
||||||
|
|
||||||
|
**Community Impact**: Demonstrating a pattern of violation of community
|
||||||
|
standards, including sustained inappropriate behavior, harassment of an
|
||||||
|
individual, or aggression toward or disparagement of classes of individuals.
|
||||||
|
|
||||||
|
**Consequence**: A permanent ban from any sort of public interaction within
|
||||||
|
the community.
|
||||||
|
|
||||||
|
## Attribution
|
||||||
|
|
||||||
|
This Code of Conduct is adapted from the [Contributor Covenant][homepage],
|
||||||
|
version 2.0, available at
|
||||||
|
https://www.contributor-covenant.org/version/2/0/code_of_conduct.html.
|
||||||
|
|
||||||
|
Community Impact Guidelines were inspired by [Mozilla's code of conduct
|
||||||
|
enforcement ladder](https://github.com/mozilla/diversity).
|
||||||
|
|
||||||
|
[homepage]: https://www.contributor-covenant.org
|
||||||
|
|
||||||
|
For answers to common questions about this code of conduct, see the FAQ at
|
||||||
|
https://www.contributor-covenant.org/faq. Translations are available at
|
||||||
|
https://www.contributor-covenant.org/translations.
|
||||||
@ -0,0 +1,19 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
import random
|
||||||
|
import os
|
||||||
|
from gtts import gTTS
|
||||||
|
|
||||||
|
max_chars = 0
|
||||||
|
|
||||||
|
|
||||||
|
class GTTS:
|
||||||
|
def __init__(self):
|
||||||
|
self.max_chars = 0
|
||||||
|
self.voices = []
|
||||||
|
|
||||||
|
def run(self, text, filepath):
|
||||||
|
tts = gTTS(text=text, lang=os.getenv("POSTLANG") or "en", slow=False)
|
||||||
|
tts.save(filepath)
|
||||||
|
|
||||||
|
def randomvoice(self):
|
||||||
|
return random.choice(self.voices)
|
||||||
@ -0,0 +1,98 @@
|
|||||||
|
import base64
|
||||||
|
import os
|
||||||
|
import random
|
||||||
|
import requests
|
||||||
|
from requests.adapters import HTTPAdapter, Retry
|
||||||
|
|
||||||
|
# from profanity_filter import ProfanityFilter
|
||||||
|
# pf = ProfanityFilter()
|
||||||
|
# Code by @JasonLovesDoggo
|
||||||
|
# https://twitter.com/scanlime/status/1512598559769702406
|
||||||
|
|
||||||
|
nonhuman = [ # DISNEY VOICES
|
||||||
|
"en_us_ghostface", # Ghost Face
|
||||||
|
"en_us_chewbacca", # Chewbacca
|
||||||
|
"en_us_c3po", # C3PO
|
||||||
|
"en_us_stitch", # Stitch
|
||||||
|
"en_us_stormtrooper", # Stormtrooper
|
||||||
|
"en_us_rocket", # Rocket
|
||||||
|
# ENGLISH VOICES
|
||||||
|
]
|
||||||
|
human = [
|
||||||
|
"en_au_001", # English AU - Female
|
||||||
|
"en_au_002", # English AU - Male
|
||||||
|
"en_uk_001", # English UK - Male 1
|
||||||
|
"en_uk_003", # English UK - Male 2
|
||||||
|
"en_us_001", # English US - Female (Int. 1)
|
||||||
|
"en_us_002", # English US - Female (Int. 2)
|
||||||
|
"en_us_006", # English US - Male 1
|
||||||
|
"en_us_007", # English US - Male 2
|
||||||
|
"en_us_009", # English US - Male 3
|
||||||
|
"en_us_010",
|
||||||
|
]
|
||||||
|
voices = nonhuman + human
|
||||||
|
|
||||||
|
noneng = [
|
||||||
|
"fr_001", # French - Male 1
|
||||||
|
"fr_002", # French - Male 2
|
||||||
|
"de_001", # German - Female
|
||||||
|
"de_002", # German - Male
|
||||||
|
"es_002", # Spanish - Male
|
||||||
|
# AMERICA VOICES
|
||||||
|
"es_mx_002", # Spanish MX - Male
|
||||||
|
"br_001", # Portuguese BR - Female 1
|
||||||
|
"br_003", # Portuguese BR - Female 2
|
||||||
|
"br_004", # Portuguese BR - Female 3
|
||||||
|
"br_005", # Portuguese BR - Male
|
||||||
|
# ASIA VOICES
|
||||||
|
"id_001", # Indonesian - Female
|
||||||
|
"jp_001", # Japanese - Female 1
|
||||||
|
"jp_003", # Japanese - Female 2
|
||||||
|
"jp_005", # Japanese - Female 3
|
||||||
|
"jp_006", # Japanese - Male
|
||||||
|
"kr_002", # Korean - Male 1
|
||||||
|
"kr_003", # Korean - Female
|
||||||
|
"kr_004", # Korean - Male 2
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
# good_voices = {'good': ['en_us_002', 'en_us_006'],
|
||||||
|
# 'ok': ['en_au_002', 'en_uk_001']} # less en_us_stormtrooper more less en_us_rocket en_us_ghostface
|
||||||
|
|
||||||
|
|
||||||
|
class TikTok: # TikTok Text-to-Speech Wrapper
|
||||||
|
def __init__(self):
|
||||||
|
self.URI_BASE = (
|
||||||
|
"https://api16-normal-useast5.us.tiktokv.com/media/api/text/speech/invoke/?text_speaker="
|
||||||
|
)
|
||||||
|
self.max_chars = 300
|
||||||
|
self.voices = {"human": human, "nonhuman": nonhuman, "noneng": noneng}
|
||||||
|
|
||||||
|
def run(self, text, filepath, random_voice: bool = False):
|
||||||
|
# if censor:
|
||||||
|
# req_text = pf.censor(req_text)
|
||||||
|
# pass
|
||||||
|
voice = (
|
||||||
|
self.randomvoice()
|
||||||
|
if random_voice
|
||||||
|
else (os.getenv("TIKTOK_VOICE") or random.choice(self.voices["human"]))
|
||||||
|
)
|
||||||
|
try:
|
||||||
|
r = requests.post(f"{self.URI_BASE}{voice}&req_text={text}&speaker_map_type=0")
|
||||||
|
except requests.exceptions.SSLError:
|
||||||
|
# https://stackoverflow.com/a/47475019/18516611
|
||||||
|
session = requests.Session()
|
||||||
|
retry = Retry(connect=3, backoff_factor=0.5)
|
||||||
|
adapter = HTTPAdapter(max_retries=retry)
|
||||||
|
session.mount("http://", adapter)
|
||||||
|
session.mount("https://", adapter)
|
||||||
|
r = session.post(f"{self.URI_BASE}{voice}&req_text={text}&speaker_map_type=0")
|
||||||
|
# print(r.text)
|
||||||
|
vstr = [r.json()["data"]["v_str"]][0]
|
||||||
|
b64d = base64.b64decode(vstr)
|
||||||
|
|
||||||
|
with open(filepath, "wb") as out:
|
||||||
|
out.write(b64d)
|
||||||
|
|
||||||
|
def randomvoice(self):
|
||||||
|
return random.choice(self.voices["human"])
|
||||||
@ -0,0 +1,66 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
from boto3 import Session
|
||||||
|
from botocore.exceptions import BotoCoreError, ClientError
|
||||||
|
import sys
|
||||||
|
import os
|
||||||
|
import random
|
||||||
|
|
||||||
|
voices = [
|
||||||
|
"Brian",
|
||||||
|
"Emma",
|
||||||
|
"Russell",
|
||||||
|
"Joey",
|
||||||
|
"Matthew",
|
||||||
|
"Joanna",
|
||||||
|
"Kimberly",
|
||||||
|
"Amy",
|
||||||
|
"Geraint",
|
||||||
|
"Nicole",
|
||||||
|
"Justin",
|
||||||
|
"Ivy",
|
||||||
|
"Kendra",
|
||||||
|
"Salli",
|
||||||
|
"Raveena",
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
class AWSPolly:
|
||||||
|
def __init__(self):
|
||||||
|
self.max_chars = 0
|
||||||
|
self.voices = voices
|
||||||
|
|
||||||
|
def run(self, text, filepath, random_voice: bool = False):
|
||||||
|
session = Session(profile_name="polly")
|
||||||
|
polly = session.client("polly")
|
||||||
|
if random_voice:
|
||||||
|
voice = self.randomvoice()
|
||||||
|
else:
|
||||||
|
if not os.getenv("AWS_VOICE"):
|
||||||
|
return ValueError(
|
||||||
|
f"Please set the environment variable AWS_VOICE to a valid voice. options are: {voices}"
|
||||||
|
)
|
||||||
|
voice = str(os.getenv("AWS_VOICE")).capitalize()
|
||||||
|
try:
|
||||||
|
# Request speech synthesis
|
||||||
|
response = polly.synthesize_speech(
|
||||||
|
Text=text, OutputFormat="mp3", VoiceId=voice, Engine="neural"
|
||||||
|
)
|
||||||
|
except (BotoCoreError, ClientError) as error:
|
||||||
|
# The service returned an error, exit gracefully
|
||||||
|
print(error)
|
||||||
|
sys.exit(-1)
|
||||||
|
|
||||||
|
# Access the audio stream from the response
|
||||||
|
if "AudioStream" in response:
|
||||||
|
file = open(filepath, "wb")
|
||||||
|
file.write(response["AudioStream"].read())
|
||||||
|
file.close()
|
||||||
|
# print_substep(f"Saved Text {idx} to MP3 files successfully.", style="bold green")
|
||||||
|
|
||||||
|
else:
|
||||||
|
# The response didn't contain audio data, exit gracefully
|
||||||
|
print("Could not stream audio")
|
||||||
|
sys.exit(-1)
|
||||||
|
|
||||||
|
def randomvoice(self):
|
||||||
|
return random.choice(self.voices)
|
||||||
@ -0,0 +1,118 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
from pathlib import Path
|
||||||
|
from typing import Tuple
|
||||||
|
import re
|
||||||
|
from os import getenv
|
||||||
|
|
||||||
|
# import sox
|
||||||
|
# from mutagen import MutagenError
|
||||||
|
# from mutagen.mp3 import MP3, HeaderNotFoundError
|
||||||
|
import translators as ts
|
||||||
|
from rich.progress import track
|
||||||
|
from moviepy.editor import AudioFileClip, CompositeAudioClip, concatenate_audioclips
|
||||||
|
from utils.console import print_step, print_substep
|
||||||
|
from utils.voice import sanitize_text
|
||||||
|
|
||||||
|
DEFUALT_MAX_LENGTH: int = 50 # video length variable
|
||||||
|
|
||||||
|
|
||||||
|
class TTSEngine:
|
||||||
|
|
||||||
|
"""Calls the given TTS engine to reduce code duplication and allow multiple TTS engines.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
tts_module : The TTS module. Your module should handle the TTS itself and saving to the given path under the run method.
|
||||||
|
reddit_object : The reddit object that contains the posts to read.
|
||||||
|
path (Optional) : The unix style path to save the mp3 files to. This must not have leading or trailing slashes.
|
||||||
|
max_length (Optional) : The maximum length of the mp3 files in total.
|
||||||
|
|
||||||
|
Notes:
|
||||||
|
tts_module must take the arguments text and filepath.
|
||||||
|
"""
|
||||||
|
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
tts_module,
|
||||||
|
reddit_object: dict,
|
||||||
|
path: str = "assets/temp/mp3",
|
||||||
|
max_length: int = DEFUALT_MAX_LENGTH,
|
||||||
|
):
|
||||||
|
self.tts_module = tts_module()
|
||||||
|
self.reddit_object = reddit_object
|
||||||
|
self.path = path
|
||||||
|
self.max_length = max_length
|
||||||
|
self.length = 0
|
||||||
|
|
||||||
|
def run(self) -> Tuple[int, int]:
|
||||||
|
|
||||||
|
Path(self.path).mkdir(parents=True, exist_ok=True)
|
||||||
|
|
||||||
|
# This file needs to be removed in case this post does not use post text, so that it wont appear in the final video
|
||||||
|
try:
|
||||||
|
Path(f"{self.path}/posttext.mp3").unlink()
|
||||||
|
except OSError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
print_step("Saving Text to MP3 files...")
|
||||||
|
|
||||||
|
self.call_tts("title", self.reddit_object["thread_title"])
|
||||||
|
if self.reddit_object["thread_post"] != "" and getenv("STORYMODE", "").casefold() == "true":
|
||||||
|
self.call_tts("posttext", self.reddit_object["thread_post"])
|
||||||
|
|
||||||
|
idx = None
|
||||||
|
for idx, comment in track(enumerate(self.reddit_object["comments"]), "Saving..."):
|
||||||
|
# ! Stop creating mp3 files if the length is greater than max length.
|
||||||
|
if self.length > self.max_length:
|
||||||
|
break
|
||||||
|
if not self.tts_module.max_chars:
|
||||||
|
self.call_tts(f"{idx}", comment["comment_body"])
|
||||||
|
else:
|
||||||
|
self.split_post(comment["comment_body"], idx)
|
||||||
|
|
||||||
|
print_substep("Saved Text to MP3 files successfully.", style="bold green")
|
||||||
|
return self.length, idx
|
||||||
|
|
||||||
|
def split_post(self, text: str, idx: int):
|
||||||
|
split_files = []
|
||||||
|
split_text = [
|
||||||
|
x.group().strip()
|
||||||
|
for x in re.finditer(rf" *((.{{0,{self.tts_module.max_chars}}})(\.|.$))", text)
|
||||||
|
]
|
||||||
|
|
||||||
|
idy = None
|
||||||
|
for idy, text_cut in enumerate(split_text):
|
||||||
|
# print(f"{idx}-{idy}: {text_cut}\n")
|
||||||
|
self.call_tts(f"{idx}-{idy}.part", text_cut)
|
||||||
|
split_files.append(AudioFileClip(f"{self.path}/{idx}-{idy}.part.mp3"))
|
||||||
|
CompositeAudioClip([concatenate_audioclips(split_files)]).write_audiofile(
|
||||||
|
f"{self.path}/{idx}.mp3", fps=44100, verbose=False, logger=None
|
||||||
|
)
|
||||||
|
|
||||||
|
for i in split_files:
|
||||||
|
name = i.filename
|
||||||
|
i.close()
|
||||||
|
Path(name).unlink()
|
||||||
|
|
||||||
|
# for i in range(0, idy + 1):
|
||||||
|
# print(f"Cleaning up {self.path}/{idx}-{i}.part.mp3")
|
||||||
|
|
||||||
|
# Path(f"{self.path}/{idx}-{i}.part.mp3").unlink()
|
||||||
|
|
||||||
|
def call_tts(self, filename: str, text: str):
|
||||||
|
self.tts_module.run(text=process_text(text), filepath=f"{self.path}/{filename}.mp3")
|
||||||
|
# try:
|
||||||
|
# self.length += MP3(f"{self.path}/{filename}.mp3").info.length
|
||||||
|
# except (MutagenError, HeaderNotFoundError):
|
||||||
|
# self.length += sox.file_info.duration(f"{self.path}/{filename}.mp3")
|
||||||
|
clip = AudioFileClip(f"{self.path}/{filename}.mp3")
|
||||||
|
self.length += clip.duration
|
||||||
|
clip.close()
|
||||||
|
|
||||||
|
def process_text(text: str):
|
||||||
|
lang = getenv("POSTLANG", "")
|
||||||
|
new_text = sanitize_text(text)
|
||||||
|
if lang:
|
||||||
|
print_substep("Translating Text...")
|
||||||
|
translated_text = ts.google(text, to_language=lang)
|
||||||
|
new_text = sanitize_text(translated_text)
|
||||||
|
return new_text
|
||||||
@ -0,0 +1,60 @@
|
|||||||
|
import random
|
||||||
|
import os
|
||||||
|
import requests
|
||||||
|
from requests.exceptions import JSONDecodeError
|
||||||
|
|
||||||
|
voices = [
|
||||||
|
"Brian",
|
||||||
|
"Emma",
|
||||||
|
"Russell",
|
||||||
|
"Joey",
|
||||||
|
"Matthew",
|
||||||
|
"Joanna",
|
||||||
|
"Kimberly",
|
||||||
|
"Amy",
|
||||||
|
"Geraint",
|
||||||
|
"Nicole",
|
||||||
|
"Justin",
|
||||||
|
"Ivy",
|
||||||
|
"Kendra",
|
||||||
|
"Salli",
|
||||||
|
"Raveena",
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
# valid voices https://lazypy.ro/tts/
|
||||||
|
|
||||||
|
|
||||||
|
class StreamlabsPolly:
|
||||||
|
def __init__(self):
|
||||||
|
self.url = "https://streamlabs.com/polly/speak"
|
||||||
|
self.max_chars = 550
|
||||||
|
self.voices = voices
|
||||||
|
|
||||||
|
def run(self, text, filepath, random_voice: bool = False):
|
||||||
|
if random_voice:
|
||||||
|
voice = self.randomvoice()
|
||||||
|
else:
|
||||||
|
if not os.getenv("STREAMLABS_VOICE"):
|
||||||
|
return ValueError(
|
||||||
|
f"Please set the environment variable STREAMLABS_VOICE to a valid voice. options are: {voices}"
|
||||||
|
)
|
||||||
|
voice = str(os.getenv("STREAMLABS_VOICE")).capitalize()
|
||||||
|
body = {"voice": voice, "text": text, "service": "polly"}
|
||||||
|
response = requests.post(self.url, data=body)
|
||||||
|
try:
|
||||||
|
voice_data = requests.get(response.json()["speak_url"])
|
||||||
|
with open(filepath, "wb") as f:
|
||||||
|
f.write(voice_data.content)
|
||||||
|
except (KeyError, JSONDecodeError):
|
||||||
|
try:
|
||||||
|
if response.json()["error"] == "No text specified!":
|
||||||
|
raise ValueError("Please specify a text to convert to speech.")
|
||||||
|
except (KeyError, JSONDecodeError):
|
||||||
|
print("Error occurred calling Streamlabs Polly")
|
||||||
|
|
||||||
|
def randomvoice(self):
|
||||||
|
return random.choice(self.voices)
|
||||||
|
|
||||||
|
|
||||||
|
# StreamlabsPolly().run(text=str('hi hi ' * 92)[1:], filepath='hello.mp3', random_voice=True)
|
||||||
Binary file not shown.
Binary file not shown.
@ -1,8 +1,12 @@
|
|||||||
|
boto3==1.24.12
|
||||||
|
botocore==1.27.22
|
||||||
gTTS==2.2.4
|
gTTS==2.2.4
|
||||||
moviepy==1.0.3
|
moviepy==1.0.3
|
||||||
mutagen==1.45.1
|
mutagen==1.45.1
|
||||||
playwright==1.22.0
|
playwright==1.23.0
|
||||||
praw==7.6.0
|
praw==7.6.0
|
||||||
python-dotenv==0.20.0
|
python-dotenv==0.20.0
|
||||||
|
pytube==12.1.0
|
||||||
|
requests==2.28.1
|
||||||
rich==12.4.4
|
rich==12.4.4
|
||||||
yt_dlp==2022.5.18
|
translators==5.3.1
|
||||||
|
|||||||
@ -1,209 +0,0 @@
|
|||||||
#!/usr/bin/env python3
|
|
||||||
# Setup Script for RedditVideoMakerBot
|
|
||||||
|
|
||||||
|
|
||||||
# Imports
|
|
||||||
import os
|
|
||||||
import subprocess
|
|
||||||
import re
|
|
||||||
from utils.console import print_markdown
|
|
||||||
from utils.console import print_step
|
|
||||||
from rich.console import Console
|
|
||||||
from utils.loader import Loader
|
|
||||||
|
|
||||||
console = Console()
|
|
||||||
|
|
||||||
|
|
||||||
def handle_input(
|
|
||||||
message: str = "",
|
|
||||||
check_type=False,
|
|
||||||
match: str = "",
|
|
||||||
err_message: str = "",
|
|
||||||
nmin=None,
|
|
||||||
nmax=None,
|
|
||||||
oob_error="",
|
|
||||||
):
|
|
||||||
match = re.compile(match + "$")
|
|
||||||
while True:
|
|
||||||
user_input = input(message + "\n> ").strip()
|
|
||||||
if re.match(match, user_input) is not None:
|
|
||||||
if check_type is not False:
|
|
||||||
try:
|
|
||||||
user_input = check_type(user_input)
|
|
||||||
if nmin is not None and user_input < nmin:
|
|
||||||
console.log("[red]" + oob_error) # Input too low failstate
|
|
||||||
continue
|
|
||||||
if nmax is not None and user_input > nmax:
|
|
||||||
console.log("[red]" + oob_error) # Input too high
|
|
||||||
continue
|
|
||||||
break # Successful type conversion and number in bounds
|
|
||||||
except ValueError:
|
|
||||||
console.log("[red]" + err_message) # Type conversion failed
|
|
||||||
continue
|
|
||||||
if (
|
|
||||||
nmin is not None and len(user_input) < nmin
|
|
||||||
): # Check if string is long enough
|
|
||||||
console.log("[red]" + oob_error)
|
|
||||||
continue
|
|
||||||
if (
|
|
||||||
nmax is not None and len(user_input) > nmax
|
|
||||||
): # Check if string is not too long
|
|
||||||
console.log("[red]" + oob_error)
|
|
||||||
continue
|
|
||||||
break
|
|
||||||
console.log("[red]" + err_message)
|
|
||||||
|
|
||||||
return user_input
|
|
||||||
|
|
||||||
|
|
||||||
if os.path.isfile(".setup-done-before"):
|
|
||||||
console.log(
|
|
||||||
"[red]Setup was already completed! Please make sure you have to run this script again. If that is such, delete the file .setup-done-before"
|
|
||||||
)
|
|
||||||
exit()
|
|
||||||
|
|
||||||
# These lines ensure the user:
|
|
||||||
# - knows they are in setup mode
|
|
||||||
# - knows that they are about to erase any other setup files/data.
|
|
||||||
|
|
||||||
print_step("Setup Assistant")
|
|
||||||
print_markdown(
|
|
||||||
"### You're in the setup wizard. Ensure you're supposed to be here, then type yes to continue. If you're not sure, type no to quit."
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
# This Input is used to ensure the user is sure they want to continue.
|
|
||||||
if input("Are you sure you want to continue? > ").strip().casefold() != "yes":
|
|
||||||
console.print("[red]Exiting...")
|
|
||||||
exit()
|
|
||||||
# This code is inaccessible if the prior check fails, and thus an else statement is unnecessary
|
|
||||||
|
|
||||||
|
|
||||||
# Again, let them know they are about to erase all other setup data.
|
|
||||||
console.print(
|
|
||||||
"[bold red] This will overwrite your current settings. Are you sure you want to continue? [bold green]yes/no"
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
if input("Are you sure you want to continue? > ").strip().casefold() != "yes":
|
|
||||||
console.print("[red]Abort mission! Exiting...")
|
|
||||||
exit()
|
|
||||||
# This is once again inaccessible if the prior checks fail
|
|
||||||
# Once they confirm, move on with the script.
|
|
||||||
console.print("[bold green]Alright! Let's get started!")
|
|
||||||
|
|
||||||
print("\n")
|
|
||||||
console.log("Ensure you have the following ready to enter:")
|
|
||||||
console.log("[bold green]Reddit Client ID")
|
|
||||||
console.log("[bold green]Reddit Client Secret")
|
|
||||||
console.log("[bold green]Reddit Username")
|
|
||||||
console.log("[bold green]Reddit Password")
|
|
||||||
console.log("[bold green]Reddit 2FA (yes or no)")
|
|
||||||
console.log("[bold green]Opacity (range of 0-1, decimals are OK)")
|
|
||||||
console.log("[bold green]Subreddit (without r/ or /r/)")
|
|
||||||
console.log("[bold green]Theme (light or dark)")
|
|
||||||
console.print(
|
|
||||||
"[green]If you don't have these, please follow the instructions in the README.md file to set them up."
|
|
||||||
)
|
|
||||||
console.print(
|
|
||||||
"[green]If you do have these, type yes to continue. If you dont, go ahead and grab those quickly and come back."
|
|
||||||
)
|
|
||||||
print()
|
|
||||||
|
|
||||||
|
|
||||||
if input("Are you sure you have the credentials? > ").strip().casefold() != "yes":
|
|
||||||
console.print("[red]I don't understand that.")
|
|
||||||
console.print("[red]Exiting...")
|
|
||||||
exit()
|
|
||||||
|
|
||||||
|
|
||||||
console.print("[bold green]Alright! Let's get started!")
|
|
||||||
|
|
||||||
# Begin the setup process.
|
|
||||||
|
|
||||||
console.log("Enter your credentials now.")
|
|
||||||
client_id = handle_input(
|
|
||||||
"Client ID > ",
|
|
||||||
False,
|
|
||||||
"[-a-zA-Z0-9._~+/]+=*",
|
|
||||||
"That is somehow not a correct ID, try again.",
|
|
||||||
12,
|
|
||||||
30,
|
|
||||||
"The ID should be over 12 and under 30 characters, double check your input.",
|
|
||||||
)
|
|
||||||
client_sec = handle_input(
|
|
||||||
"Client Secret > ",
|
|
||||||
False,
|
|
||||||
"[-a-zA-Z0-9._~+/]+=*",
|
|
||||||
"That is somehow not a correct secret, try again.",
|
|
||||||
20,
|
|
||||||
40,
|
|
||||||
"The secret should be over 20 and under 40 characters, double check your input.",
|
|
||||||
)
|
|
||||||
user = handle_input(
|
|
||||||
"Username > ",
|
|
||||||
False,
|
|
||||||
r"[_0-9a-zA-Z]+",
|
|
||||||
"That is not a valid user",
|
|
||||||
3,
|
|
||||||
20,
|
|
||||||
"A username HAS to be between 3 and 20 characters",
|
|
||||||
)
|
|
||||||
passw = handle_input("Password > ", False, ".*", "", 8, None, "Password too short")
|
|
||||||
twofactor = handle_input(
|
|
||||||
"2fa Enabled? (yes/no) > ",
|
|
||||||
False,
|
|
||||||
r"(yes)|(no)",
|
|
||||||
"You need to input either yes or no",
|
|
||||||
)
|
|
||||||
opacity = handle_input(
|
|
||||||
"Opacity? (range of 0-1) > ",
|
|
||||||
float,
|
|
||||||
".*",
|
|
||||||
"You need to input a number between 0 and 1",
|
|
||||||
0,
|
|
||||||
1,
|
|
||||||
"Your number is not between 0 and 1",
|
|
||||||
)
|
|
||||||
subreddit = handle_input(
|
|
||||||
"Subreddit (without r/) > ",
|
|
||||||
False,
|
|
||||||
r"[_0-9a-zA-Z]+",
|
|
||||||
"This subreddit cannot exist, make sure you typed it in correctly and removed the r/ (or /r/).",
|
|
||||||
3,
|
|
||||||
20,
|
|
||||||
"A subreddit name HAS to be between 3 and 20 characters",
|
|
||||||
)
|
|
||||||
theme = handle_input(
|
|
||||||
"Theme? (light or dark) > ",
|
|
||||||
False,
|
|
||||||
r"(light)|(dark)",
|
|
||||||
"You need to input 'light' or 'dark'",
|
|
||||||
)
|
|
||||||
loader = Loader("Attempting to save your credentials...", "Done!").start()
|
|
||||||
# you can also put a while loop here, e.g. while VideoIsBeingMade == True: ...
|
|
||||||
console.log("Writing to the .env file...")
|
|
||||||
with open(".env", "w") as f:
|
|
||||||
f.write(
|
|
||||||
f"""REDDIT_CLIENT_ID="{client_id}"
|
|
||||||
REDDIT_CLIENT_SECRET="{client_sec}"
|
|
||||||
REDDIT_USERNAME="{user}"
|
|
||||||
REDDIT_PASSWORD="{passw}"
|
|
||||||
REDDIT_2FA="{twofactor}"
|
|
||||||
THEME="{theme}"
|
|
||||||
SUBREDDIT="{subreddit}"
|
|
||||||
OPACITY={opacity}
|
|
||||||
"""
|
|
||||||
)
|
|
||||||
|
|
||||||
with open(".setup-done-before", "w") as f:
|
|
||||||
f.write(
|
|
||||||
"This file blocks the setup assistant from running again. Delete this file to run setup again."
|
|
||||||
)
|
|
||||||
|
|
||||||
loader.stop()
|
|
||||||
|
|
||||||
console.log("[bold green]Setup Complete! Returning...")
|
|
||||||
|
|
||||||
# Post-Setup: send message and try to run main.py again.
|
|
||||||
subprocess.call("python3 main.py", shell=True)
|
|
||||||
@ -0,0 +1,193 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
import os
|
||||||
|
from rich.console import Console
|
||||||
|
from rich.table import Table
|
||||||
|
from rich import box
|
||||||
|
import re
|
||||||
|
import dotenv
|
||||||
|
from utils.console import handle_input
|
||||||
|
|
||||||
|
console = Console()
|
||||||
|
|
||||||
|
|
||||||
|
def check_env() -> bool:
|
||||||
|
"""Checks to see what's been put in .env
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
bool: Whether or not everything was put in properly
|
||||||
|
"""
|
||||||
|
if not os.path.exists(".env.template"):
|
||||||
|
console.print("[red]Couldn't find .env.template. Unable to check variables.")
|
||||||
|
return True
|
||||||
|
if not os.path.exists(".env"):
|
||||||
|
console.print("[red]Couldn't find the .env file, creating one now.")
|
||||||
|
with open(".env", "x", encoding="utf-8") as file:
|
||||||
|
file.write("")
|
||||||
|
success = True
|
||||||
|
with open(".env.template", "r", encoding="utf-8") as template:
|
||||||
|
# req_envs = [env.split("=")[0] for env in template.readlines() if "=" in env]
|
||||||
|
matching = {}
|
||||||
|
explanations = {}
|
||||||
|
bounds = {}
|
||||||
|
types = {}
|
||||||
|
oob_errors = {}
|
||||||
|
examples = {}
|
||||||
|
req_envs = []
|
||||||
|
var_optional = False
|
||||||
|
for line in template.readlines():
|
||||||
|
if line.startswith("#") is not True and "=" in line and var_optional is not True:
|
||||||
|
req_envs.append(line.split("=")[0])
|
||||||
|
if "#" in line:
|
||||||
|
examples[line.split("=")[0]] = "#".join(line.split("#")[1:]).strip()
|
||||||
|
elif "#OPTIONAL" in line:
|
||||||
|
var_optional = True
|
||||||
|
elif line.startswith("#MATCH_REGEX "):
|
||||||
|
matching[req_envs[-1]] = line.removeprefix("#MATCH_REGEX ")[:-1]
|
||||||
|
var_optional = False
|
||||||
|
elif line.startswith("#OOB_ERROR "):
|
||||||
|
oob_errors[req_envs[-1]] = line.removeprefix("#OOB_ERROR ")[:-1]
|
||||||
|
var_optional = False
|
||||||
|
elif line.startswith("#RANGE "):
|
||||||
|
bounds[req_envs[-1]] = tuple(
|
||||||
|
map(
|
||||||
|
lambda x: float(x) if x != "None" else None,
|
||||||
|
line.removeprefix("#RANGE ")[:-1].split(":"),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
var_optional = False
|
||||||
|
elif line.startswith("#MATCH_TYPE "):
|
||||||
|
types[req_envs[-1]] = eval(line.removeprefix("#MATCH_TYPE ")[:-1].split()[0])
|
||||||
|
var_optional = False
|
||||||
|
elif line.startswith("#EXPLANATION "):
|
||||||
|
explanations[req_envs[-1]] = line.removeprefix("#EXPLANATION ")[:-1]
|
||||||
|
var_optional = False
|
||||||
|
else:
|
||||||
|
var_optional = False
|
||||||
|
missing = set()
|
||||||
|
incorrect = set()
|
||||||
|
dotenv.load_dotenv()
|
||||||
|
for env in req_envs:
|
||||||
|
value = os.getenv(env)
|
||||||
|
if value is None:
|
||||||
|
missing.add(env)
|
||||||
|
continue
|
||||||
|
if env in matching.keys():
|
||||||
|
re.match(matching[env], value) is None and incorrect.add(env)
|
||||||
|
if env in bounds.keys() and env not in types.keys():
|
||||||
|
len(value) >= bounds[env][0] or (
|
||||||
|
len(bounds[env]) > 1 and bounds[env][1] >= len(value)
|
||||||
|
) or incorrect.add(env)
|
||||||
|
continue
|
||||||
|
if env in types.keys():
|
||||||
|
try:
|
||||||
|
temp = types[env](value)
|
||||||
|
if env in bounds.keys():
|
||||||
|
(bounds[env][0] <= temp or incorrect.add(env)) and len(bounds[env]) > 1 and (
|
||||||
|
bounds[env][1] >= temp or incorrect.add(env)
|
||||||
|
)
|
||||||
|
except ValueError:
|
||||||
|
incorrect.add(env)
|
||||||
|
|
||||||
|
if len(missing):
|
||||||
|
table = Table(
|
||||||
|
title="Missing variables",
|
||||||
|
highlight=True,
|
||||||
|
show_lines=True,
|
||||||
|
box=box.ROUNDED,
|
||||||
|
border_style="#414868",
|
||||||
|
header_style="#C0CAF5 bold",
|
||||||
|
title_justify="left",
|
||||||
|
title_style="#C0CAF5 bold",
|
||||||
|
)
|
||||||
|
table.add_column("Variable", justify="left", style="#7AA2F7 bold", no_wrap=True)
|
||||||
|
table.add_column("Explanation", justify="left", style="#BB9AF7", no_wrap=False)
|
||||||
|
table.add_column("Example", justify="center", style="#F7768E", no_wrap=True)
|
||||||
|
table.add_column("Min", justify="right", style="#F7768E", no_wrap=True)
|
||||||
|
table.add_column("Max", justify="left", style="#F7768E", no_wrap=True)
|
||||||
|
for env in missing:
|
||||||
|
table.add_row(
|
||||||
|
env,
|
||||||
|
explanations[env] if env in explanations.keys() else "No explanation given",
|
||||||
|
examples[env] if env in examples.keys() else "",
|
||||||
|
str(bounds[env][0]) if env in bounds.keys() and bounds[env][1] is not None else "",
|
||||||
|
str(bounds[env][1])
|
||||||
|
if env in bounds.keys() and len(bounds[env]) > 1 and bounds[env][1] is not None
|
||||||
|
else "",
|
||||||
|
)
|
||||||
|
console.print(table)
|
||||||
|
success = False
|
||||||
|
if len(incorrect):
|
||||||
|
table = Table(
|
||||||
|
title="Incorrect variables",
|
||||||
|
highlight=True,
|
||||||
|
show_lines=True,
|
||||||
|
box=box.ROUNDED,
|
||||||
|
border_style="#414868",
|
||||||
|
header_style="#C0CAF5 bold",
|
||||||
|
title_justify="left",
|
||||||
|
title_style="#C0CAF5 bold",
|
||||||
|
)
|
||||||
|
table.add_column("Variable", justify="left", style="#7AA2F7 bold", no_wrap=True)
|
||||||
|
table.add_column("Current value", justify="left", style="#F7768E", no_wrap=False)
|
||||||
|
table.add_column("Explanation", justify="left", style="#BB9AF7", no_wrap=False)
|
||||||
|
table.add_column("Example", justify="center", style="#F7768E", no_wrap=True)
|
||||||
|
table.add_column("Min", justify="right", style="#F7768E", no_wrap=True)
|
||||||
|
table.add_column("Max", justify="left", style="#F7768E", no_wrap=True)
|
||||||
|
for env in incorrect:
|
||||||
|
table.add_row(
|
||||||
|
env,
|
||||||
|
os.getenv(env),
|
||||||
|
explanations[env] if env in explanations.keys() else "No explanation given",
|
||||||
|
str(types[env].__name__) if env in types.keys() else "str",
|
||||||
|
str(bounds[env][0]) if env in bounds.keys() else "None",
|
||||||
|
str(bounds[env][1]) if env in bounds.keys() and len(bounds[env]) > 1 else "None",
|
||||||
|
)
|
||||||
|
missing.add(env)
|
||||||
|
console.print(table)
|
||||||
|
success = False
|
||||||
|
if success is True:
|
||||||
|
return True
|
||||||
|
console.print(
|
||||||
|
"[green]Do you want to automatically overwrite incorrect variables and add the missing variables? (y/n)"
|
||||||
|
)
|
||||||
|
if not input().casefold().startswith("y"):
|
||||||
|
console.print("[red]Aborting: Unresolved missing variables")
|
||||||
|
return False
|
||||||
|
if len(incorrect):
|
||||||
|
with open(".env", "r+", encoding="utf-8") as env_file:
|
||||||
|
lines = []
|
||||||
|
for line in env_file.readlines():
|
||||||
|
line.split("=")[0].strip() not in incorrect and lines.append(line)
|
||||||
|
env_file.seek(0)
|
||||||
|
env_file.write("\n".join(lines))
|
||||||
|
env_file.truncate()
|
||||||
|
console.print("[green]Successfully removed incorrectly set variables from .env")
|
||||||
|
with open(".env", "a", encoding="utf-8") as env_file:
|
||||||
|
for env in missing:
|
||||||
|
env_file.write(
|
||||||
|
env
|
||||||
|
+ "="
|
||||||
|
+ ('"' if env not in types.keys() else "")
|
||||||
|
+ str(
|
||||||
|
handle_input(
|
||||||
|
"[#F7768E bold]" + env + "[#C0CAF5 bold]=",
|
||||||
|
types[env] if env in types.keys() else False,
|
||||||
|
matching[env] if env in matching.keys() else ".*",
|
||||||
|
explanations[env]
|
||||||
|
if env in explanations.keys()
|
||||||
|
else "Incorrect input. Try again.",
|
||||||
|
bounds[env][0] if env in bounds.keys() else None,
|
||||||
|
bounds[env][1] if env in bounds.keys() and len(bounds[env]) > 1 else None,
|
||||||
|
oob_errors[env] if env in oob_errors.keys() else "Input too long/short.",
|
||||||
|
extra_info="[#C0CAF5 bold]⮶ "
|
||||||
|
+ (explanations[env] if env in explanations.keys() else "No info available"),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
+ ('"' if env not in types.keys() else "")
|
||||||
|
+ "\n"
|
||||||
|
)
|
||||||
|
return True
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
check_env()
|
||||||
@ -0,0 +1,27 @@
|
|||||||
|
import os
|
||||||
|
from os.path import exists
|
||||||
|
|
||||||
|
|
||||||
|
def cleanup() -> int:
|
||||||
|
"""Deletes all temporary assets in assets/temp
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
int: How many files were deleted
|
||||||
|
"""
|
||||||
|
if exists("./assets/temp"):
|
||||||
|
count = 0
|
||||||
|
files = [f for f in os.listdir(".") if f.endswith(".mp4") and "temp" in f.lower()]
|
||||||
|
count += len(files)
|
||||||
|
for f in files:
|
||||||
|
os.remove(f)
|
||||||
|
try:
|
||||||
|
for file in os.listdir("./assets/temp/mp4"):
|
||||||
|
count += 1
|
||||||
|
os.remove("./assets/temp/mp4/" + file)
|
||||||
|
except FileNotFoundError:
|
||||||
|
pass
|
||||||
|
for file in os.listdir("./assets/temp/mp3"):
|
||||||
|
count += 1
|
||||||
|
os.remove("./assets/temp/mp3/" + file)
|
||||||
|
return count
|
||||||
|
return 0
|
||||||
@ -0,0 +1,46 @@
|
|||||||
|
# write a class that takes .env file and parses it into a dictionary
|
||||||
|
from dotenv import dotenv_values
|
||||||
|
|
||||||
|
DEFAULTS = {
|
||||||
|
"SUBREDDIT": "AskReddit",
|
||||||
|
"ALLOW_NSFW": "False",
|
||||||
|
"POST_ID": "",
|
||||||
|
"THEME": "DARK",
|
||||||
|
"REDDIT_2FA": "no",
|
||||||
|
"TIMES_TO_RUN": "",
|
||||||
|
"MAX_COMMENT_LENGTH": "500",
|
||||||
|
"OPACITY": "1",
|
||||||
|
"VOICE": "en_us_001",
|
||||||
|
"STORYMODE": "False",
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
class Config:
|
||||||
|
def __init__(self):
|
||||||
|
self.raw = dotenv_values("../.env")
|
||||||
|
self.load_attrs()
|
||||||
|
|
||||||
|
def __getattr__(self, attr): # code completion for attributes fix.
|
||||||
|
return getattr(self, attr)
|
||||||
|
|
||||||
|
def load_attrs(self):
|
||||||
|
for key, value in self.raw.items():
|
||||||
|
self.add_attr(key, value)
|
||||||
|
|
||||||
|
def add_attr(self, key, value):
|
||||||
|
if value is None or value == "":
|
||||||
|
setattr(self, key, DEFAULTS[key])
|
||||||
|
else:
|
||||||
|
setattr(self, key, str(value))
|
||||||
|
|
||||||
|
|
||||||
|
config = Config()
|
||||||
|
|
||||||
|
print(config.SUBREDDIT)
|
||||||
|
# def temp():
|
||||||
|
# root = ''
|
||||||
|
# if isinstance(root, praw.models.Submission):
|
||||||
|
# root_type = 'submission'
|
||||||
|
# elif isinstance(root, praw.models.Comment):
|
||||||
|
# root_type = 'comment'
|
||||||
|
#
|
||||||
@ -0,0 +1,54 @@
|
|||||||
|
import json
|
||||||
|
from os import getenv
|
||||||
|
from utils.console import print_substep
|
||||||
|
|
||||||
|
|
||||||
|
def get_subreddit_undone(submissions: list, subreddit):
|
||||||
|
"""_summary_
|
||||||
|
|
||||||
|
Args:
|
||||||
|
submissions (list): List of posts that are going to potentially be generated into a video
|
||||||
|
subreddit (praw.Reddit.SubredditHelper): Chosen subreddit
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Any: The submission that has not been done
|
||||||
|
"""
|
||||||
|
# recursively checks if the top submission in the list was already done.
|
||||||
|
|
||||||
|
with open("./video_creation/data/videos.json", "r", encoding="utf-8") as done_vids_raw:
|
||||||
|
done_videos = json.load(done_vids_raw)
|
||||||
|
for submission in submissions:
|
||||||
|
if already_done(done_videos, submission):
|
||||||
|
continue
|
||||||
|
if submission.over_18:
|
||||||
|
try:
|
||||||
|
if getenv("ALLOW_NSFW").casefold() == "false":
|
||||||
|
print_substep("NSFW Post Detected. Skipping...")
|
||||||
|
continue
|
||||||
|
except AttributeError:
|
||||||
|
print_substep("NSFW settings not defined. Skipping NSFW post...")
|
||||||
|
if submission.stickied:
|
||||||
|
print_substep("This post was pinned by moderators. Skipping...")
|
||||||
|
continue
|
||||||
|
return submission
|
||||||
|
print("all submissions have been done going by top submission order")
|
||||||
|
return get_subreddit_undone(
|
||||||
|
subreddit.top(time_filter="hour"), subreddit
|
||||||
|
) # all of the videos in hot have already been done
|
||||||
|
|
||||||
|
|
||||||
|
def already_done(done_videos: list, submission) -> bool:
|
||||||
|
"""Checks to see if the given submission is in the list of videos
|
||||||
|
|
||||||
|
Args:
|
||||||
|
done_videos (list): Finished videos
|
||||||
|
submission (Any): The submission
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Boolean: Whether the video was found in the list
|
||||||
|
"""
|
||||||
|
|
||||||
|
for video in done_videos:
|
||||||
|
if video["id"] == str(submission):
|
||||||
|
return True
|
||||||
|
return False
|
||||||
@ -0,0 +1,60 @@
|
|||||||
|
import json
|
||||||
|
import os
|
||||||
|
import time
|
||||||
|
from os import getenv
|
||||||
|
from typing import Dict
|
||||||
|
|
||||||
|
from praw.models import Submission
|
||||||
|
|
||||||
|
from utils.console import print_step
|
||||||
|
|
||||||
|
|
||||||
|
def check_done(
|
||||||
|
redditobj: Submission,
|
||||||
|
) -> Submission:
|
||||||
|
# don't set this to be run anyplace that isn't subreddit.py bc of inspect stack
|
||||||
|
"""Checks if the chosen post has already been generated
|
||||||
|
|
||||||
|
Args:
|
||||||
|
redditobj (Dict[str]): Reddit object gotten from reddit/subreddit.py
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Dict[str]|None: Reddit object in args
|
||||||
|
"""
|
||||||
|
with open("./video_creation/data/videos.json", "r", encoding="utf-8") as done_vids_raw:
|
||||||
|
done_videos = json.load(done_vids_raw)
|
||||||
|
for video in done_videos:
|
||||||
|
if video["id"] == str(redditobj):
|
||||||
|
if getenv("POST_ID"):
|
||||||
|
print_step(
|
||||||
|
"You already have done this video but since it was declared specifically in the .env file the program will continue"
|
||||||
|
)
|
||||||
|
return redditobj
|
||||||
|
print_step("Getting new post as the current one has already been done")
|
||||||
|
return None
|
||||||
|
return redditobj
|
||||||
|
|
||||||
|
|
||||||
|
def save_data(filename: str, reddit_title: str, reddit_id: str):
|
||||||
|
"""Saves the videos that have already been generated to a JSON file in video_creation/data/videos.json
|
||||||
|
|
||||||
|
Args:
|
||||||
|
filename (str): The finished video title name
|
||||||
|
@param filename:
|
||||||
|
@param reddit_id:
|
||||||
|
@param reddit_title:
|
||||||
|
"""
|
||||||
|
with open("./video_creation/data/videos.json", "r+", encoding="utf-8") as raw_vids:
|
||||||
|
done_vids = json.load(raw_vids)
|
||||||
|
if reddit_id in [video["id"] for video in done_vids]:
|
||||||
|
return # video already done but was specified to continue anyway in the .env file
|
||||||
|
payload = {
|
||||||
|
"id": reddit_id,
|
||||||
|
"time": str(int(time.time())),
|
||||||
|
"background_credit": str(os.getenv("background_credit")),
|
||||||
|
"reddit_title": reddit_title,
|
||||||
|
"filename": filename,
|
||||||
|
}
|
||||||
|
done_vids.append(payload)
|
||||||
|
raw_vids.seek(0)
|
||||||
|
json.dump(done_vids, raw_vids, ensure_ascii=False, indent=4)
|
||||||
@ -1,8 +0,0 @@
|
|||||||
[
|
|
||||||
{
|
|
||||||
"name": "USER",
|
|
||||||
"value": "eyJwcmVmcyI6eyJ0b3BDb250ZW50RGlzbWlzc2FsVGltZSI6MCwiZ2xvYmFsVGhlbWUiOiJSRURESVQiLCJuaWdodG1vZGUiOnRydWUsImNvbGxhcHNlZFRyYXlTZWN0aW9ucyI6eyJmYXZvcml0ZXMiOmZhbHNlLCJtdWx0aXMiOmZhbHNlLCJtb2RlcmF0aW5nIjpmYWxzZSwic3Vic2NyaXB0aW9ucyI6ZmFsc2UsInByb2ZpbGVzIjpmYWxzZX0sInRvcENvbnRlbnRUaW1lc0Rpc21pc3NlZCI6MH19",
|
|
||||||
"domain": ".reddit.com",
|
|
||||||
"path": "/"
|
|
||||||
}
|
|
||||||
]
|
|
||||||
@ -0,0 +1,14 @@
|
|||||||
|
[
|
||||||
|
{
|
||||||
|
"name": "USER",
|
||||||
|
"value": "eyJwcmVmcyI6eyJ0b3BDb250ZW50RGlzbWlzc2FsVGltZSI6MCwiZ2xvYmFsVGhlbWUiOiJSRURESVQiLCJuaWdodG1vZGUiOnRydWUsImNvbGxhcHNlZFRyYXlTZWN0aW9ucyI6eyJmYXZvcml0ZXMiOmZhbHNlLCJtdWx0aXMiOmZhbHNlLCJtb2RlcmF0aW5nIjpmYWxzZSwic3Vic2NyaXB0aW9ucyI6ZmFsc2UsInByb2ZpbGVzIjpmYWxzZX0sInRvcENvbnRlbnRUaW1lc0Rpc21pc3NlZCI6MH19",
|
||||||
|
"domain": ".reddit.com",
|
||||||
|
"path": "/"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "eu_cookie",
|
||||||
|
"value": "{%22opted%22:true%2C%22nonessential%22:false}",
|
||||||
|
"domain": ".reddit.com",
|
||||||
|
"path": "/"
|
||||||
|
}
|
||||||
|
]
|
||||||
@ -0,0 +1,8 @@
|
|||||||
|
[
|
||||||
|
{
|
||||||
|
"name": "eu_cookie",
|
||||||
|
"value": "{%22opted%22:true%2C%22nonessential%22:false}",
|
||||||
|
"domain": ".reddit.com",
|
||||||
|
"path": "/"
|
||||||
|
}
|
||||||
|
]
|
||||||
@ -0,0 +1 @@
|
|||||||
|
[]
|
||||||
@ -1,48 +1,57 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python
|
||||||
from gtts import gTTS
|
|
||||||
from pathlib import Path
|
|
||||||
from mutagen.mp3 import MP3
|
|
||||||
from utils.console import print_step, print_substep
|
|
||||||
from rich.progress import track
|
|
||||||
import re
|
|
||||||
|
|
||||||
|
import os
|
||||||
|
from typing import Dict, Tuple
|
||||||
|
|
||||||
def save_text_to_mp3(reddit_obj):
|
from rich.console import Console
|
||||||
"""Saves Text to MP3 files.
|
|
||||||
|
|
||||||
Args:
|
from TTS.engine_wrapper import TTSEngine
|
||||||
reddit_obj : The reddit object you received from the reddit API in the askreddit.py file.
|
from TTS.GTTS import GTTS
|
||||||
"""
|
from TTS.streamlabs_polly import StreamlabsPolly
|
||||||
print_step("Saving Text to MP3 files...")
|
from TTS.aws_polly import AWSPolly
|
||||||
length = 0
|
from TTS.TikTok import TikTok
|
||||||
|
|
||||||
|
from utils.console import print_table, print_step
|
||||||
|
|
||||||
# Create a folder for the mp3 files.
|
|
||||||
Path("assets/mp3").mkdir(parents=True, exist_ok=True)
|
|
||||||
|
|
||||||
tts = gTTS(text=reddit_obj["thread_title"], lang="en", slow=False)
|
console = Console()
|
||||||
tts.save("assets/mp3/title.mp3")
|
|
||||||
length += MP3("assets/mp3/title.mp3").info.length
|
|
||||||
|
|
||||||
try:
|
TTSProviders = {
|
||||||
Path("assets/mp3/posttext.mp3").unlink()
|
"GoogleTranslate": GTTS,
|
||||||
except OSError:
|
"AWSPolly": AWSPolly,
|
||||||
pass
|
"StreamlabsPolly": StreamlabsPolly,
|
||||||
|
"TikTok": TikTok,
|
||||||
|
}
|
||||||
|
|
||||||
if reddit_obj["thread_post"] != "":
|
|
||||||
tts = gTTS(text=reddit_obj["thread_post"], lang="en", slow=False)
|
|
||||||
tts.save("assets/mp3/posttext.mp3")
|
|
||||||
length += MP3("assets/mp3/posttext.mp3").info.length
|
|
||||||
|
|
||||||
for idx, comment in track(enumerate(reddit_obj["comments"]), "Saving..."):
|
def save_text_to_mp3(reddit_obj) -> Tuple[int, int]:
|
||||||
# ! Stop creating mp3 files if the length is greater than 50 seconds. This can be longer, but this is just a good starting point
|
"""Saves text to MP3 files.
|
||||||
if length > 50:
|
|
||||||
|
Args:
|
||||||
|
reddit_obj (dict[str]): Reddit object received from reddit API in reddit/subreddit.py
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
tuple[int,int]: (total length of the audio, the number of comments audio was generated for)
|
||||||
|
"""
|
||||||
|
|
||||||
|
env = os.getenv("TTSCHOICE", "")
|
||||||
|
if env.casefold() in map(lambda _: _.casefold(), TTSProviders):
|
||||||
|
text_to_mp3 = TTSEngine(get_case_insensitive_key_value(TTSProviders, env), reddit_obj)
|
||||||
|
else:
|
||||||
|
while True:
|
||||||
|
print_step("Please choose one of the following TTS providers: ")
|
||||||
|
print_table(TTSProviders)
|
||||||
|
choice = input("\n")
|
||||||
|
if choice.casefold() in map(lambda _: _.casefold(), TTSProviders):
|
||||||
break
|
break
|
||||||
comment=comment["comment_body"]
|
print("Unknown Choice")
|
||||||
text=re.sub('((http|https)\:\/\/)?[a-zA-Z0-9\.\/\?\:@\-_=#]+\.([a-zA-Z]){2,6}([a-zA-Z0-9\.\&\/\?\:@\-_=#])*', '', comment)
|
text_to_mp3 = TTSEngine(get_case_insensitive_key_value(TTSProviders, choice), reddit_obj)
|
||||||
tts = gTTS(text, lang="en", slow=False)
|
|
||||||
tts.save(f"assets/mp3/{idx}.mp3")
|
return text_to_mp3.run()
|
||||||
length += MP3(f"assets/mp3/{idx}.mp3").info.length
|
|
||||||
|
|
||||||
print_substep("Saved Text to MP3 files successfully.", style="bold green")
|
def get_case_insensitive_key_value(input_dict, key):
|
||||||
# ! Return the index so we know how many screenshots of comments we need to make.
|
return next(
|
||||||
return length, idx
|
(value for dict_key, value in input_dict.items() if dict_key.lower() == key.lower()),
|
||||||
|
None,
|
||||||
|
)
|
||||||
|
|||||||
Loading…
Reference in new issue