some fixes for tts, adding experimental code for gallery lists

This commit is contained in:
Believethehype 2024-06-27 21:19:04 +02:00
parent ffcedafe74
commit f7e1bc5e52
7 changed files with 61 additions and 22 deletions

View File

@ -125,15 +125,15 @@ class TextToSpeech(DVMTaskInterface):
model = "tts_models/multilingual/multi-dataset/xtts_v2"
tts = TTS(model).to(device)
text = options["prompt"]
text = options["prompt"].replace("~", "around ").replace("#", "hashtag").replace(" LFG", " Let's fucking go")
text_clean = re.sub(
r'''(?i)\b((?:https?://|www\d{0,3}[.]|[a-z0-9.\-]+[.][a-z]{2,4}/)(?:[^\s()<>]+|\(([^\s()<>]+|(\([^\s()<>]+\)))*\))+(?:\(([^\s()<>]+|(\([^\s()<>]+\)))*\)|[^\s`!()\[\]{};:'".,<>?«»“”‘’]))''',
" ", text)
tts.tts_to_file(
text=text_clean,
speaker_wav=options["input_wav"], language=options["language"], file_path="outputs/output.wav")
result = upload_media_to_hoster("outputs/output.wav")
speaker_wav=options["input_wav"], language=options["language"], file_path="outputs/output.mp3")
result = upload_media_to_hoster("outputs/output.mp3")
print(result)
return result
except Exception as e:

View File

@ -0,0 +1,16 @@
from nostr_sdk import Tag, Keys, EventBuilder, Kind
from nostr_dvm.utils.definitions import EventDefinitions
from nostr_dvm.utils.nostr_utils import send_event
from nostr_dvm.utils.print import bcolors
async def gallery_announce_list(tags, dvm_config, client):
keys = Keys.parse(dvm_config.NIP89.PK)
content = ""
event = EventBuilder(Kind(10011), content, tags).to_event(keys)
eventid = await send_event(event, client=client, dvm_config=dvm_config, blastr=True)
print(bcolors.BLUE + "[" + dvm_config.NIP89.NAME + "] Announced Gallery for " + dvm_config.NIP89.NAME +" (EventID: " + str(eventid.to_hex()) +")" + bcolors.ENDC)

View File

@ -154,33 +154,25 @@ def upload_media_to_hoster(filepath: str):
sizeinmb = file_stats.st_size / (1024 * 1024)
print("Filesize of Uploaded media: " + str(sizeinmb) + " Mb.")
if sizeinmb > 25:
print("Filesize over Nostr.build limited, using catbox")
uploader = CatboxUploader(filepath)
result = uploader.execute()
return result
else:
url = 'https://nostr.build/api/v2/upload/files'
response = requests.post(url, files=files)
print(response.text)
json_object = json.loads(response.text)
result = json_object["data"][0]["url"]
return result
except:
except Exception as e:
try:
file = {'file': open(filepath, 'rb')}
url = 'https://nostrfiles.dev/upload_image'
response = requests.post(url, files=file)
json_object = json.loads(response.text)
print(json_object["url"])
return json_object["url"]
# fallback filehoster
uploader = CatboxUploader(filepath)
result = uploader.execute()
print(result)
return result
except:
try:
uploader = CatboxUploader(filepath)
result = uploader.execute()
print(result)
return result
except:
raise Exception("Upload not possible, all hosters didn't work or couldn't generate output")
raise Exception("Upload not possible, all hosters didn't work or couldn't generate output")
def build_status_reaction(status, task, amount, content, dvm_config):

View File

@ -1,6 +1,6 @@
from setuptools import setup, find_packages
VERSION = '0.6.27'
VERSION = '0.6.28'
DESCRIPTION = 'A framework to build and run Nostr NIP90 Data Vending Machines'
LONG_DESCRIPTION = ('A framework to build and run Nostr NIP90 Data Vending Machines. See the github repository for more information')

BIN
tests/output.wav Normal file

Binary file not shown.

View File

@ -7,6 +7,9 @@ from nostr_sdk import Keys, Client, Tag, EventBuilder, Filter, HandleNotificatio
nip04_encrypt, EventId, Options, PublicKey, Event, NostrSigner, Nip19Event
from nostr_dvm.utils import definitions, dvmconfig
from nostr_dvm.utils.dvmconfig import DVMConfig
from nostr_dvm.utils.gallery_utils import gallery_announce_list
from nostr_dvm.utils.nip89_utils import NIP89Config
from nostr_dvm.utils.nostr_utils import check_and_set_private_key
@ -25,7 +28,7 @@ async def test():
for relay in relay_list:
await client.add_relay(relay)
client.connect()
await client.connect()
await test_referred_events(client,"c70fbd4dbaad22c427d4359981d3bdddd3971ed1a38227ca2f8e5e760f58103c",
definitions.EventDefinitions.ANY_RESULT)
@ -69,6 +72,31 @@ async def test_referred_events(client, event_id, kinds=None):
return None
async def test_gallery():
relay_list = dvmconfig.DVMConfig.RELAY_LIST
keys = Keys.parse(check_and_set_private_key("test_client"))
wait_for_send = False
skip_disconnected_relays = True
opts = (Options().wait_for_send(wait_for_send).send_timeout(timedelta(seconds=5))
.skip_disconnected_relays(skip_disconnected_relays))
signer = NostrSigner.keys(keys)
client = Client.with_opts(signer, opts)
for relay in relay_list:
await client.add_relay(relay)
await client.connect()
dvm_config = DVMConfig()
dvm_config.NIP89 = NIP89Config()
dvm_config.NIP89.PK = Keys.parse("nsec...").secret_key().to_hex()
tags = [Tag.parse(["d", "gallery"]),
Tag.parse(["image", "https://i.nostr.build/7G2G2.jpg"]),
Tag.parse(["image", "https://i.nostr.build/XVLkd.jpg"])]
await gallery_announce_list(tags, dvm_config, client)
async def test_search_by_user_since_days(client, pubkey, days, prompt):
since_seconds = int(days) * 24 * 60 * 60
dif = Timestamp.now().as_secs() - since_seconds
@ -95,7 +123,7 @@ if __name__ == '__main__':
else:
raise FileNotFoundError(f'.env file not found at {env_path} ')
asyncio.run(test())
asyncio.run(test_gallery())
# works

3
tests/upload_hoster.py Normal file
View File

@ -0,0 +1,3 @@
from nostr_dvm.utils.output_utils import upload_media_to_hoster
upload_media_to_hoster("tests/output.wav")