python-project/DiscordBot/madlibsDiscord.py

#!/usr/bin/python
#General purpose Discord Bot developed by Caleb Fontenot (CCF_100)
#Written in python.
# Toggle me for debugging
debug = 1
# Displays warning message
testing = 1
# Import the libraries we will use
#from mega import Mega
import tracemalloc
#from concurrent.futures import ProcessPoolExecutor
from datetime import datetime
from gtts import gTTS
import discord
from discord.ext import commands
import re
import sys
import random
import os
import asyncio
from threading import Thread
import ffmpeg
#import talkey
prefix = 'mad!'
tracemalloc.start()

async def madlibsLoop():
    # In the future we will detect what channel we were summoned in, but for now:
    activeChannel = 656233549837631508
    #Init tts, and connect voice to channel, reinit connection if broken
    voiceChannel = client.get_channel(682688245964079127)
    voice = await voiceChannel.connect()
    if voice.is_connected() == True:
        await voice.disconnect()
    voice = await voiceChannel.connect()
    # Set bot presence
    await client.change_presence(activity=discord.Game(name='madlibs.py'))
    # Introduce yourself
    channel = client.get_channel(656233549837631508)
    await channel.send("**<<madlibsDiscord.py <:python:656239601723113472> - Written by CCF_100>>**") 
    await channel.send("Initial project started on **July 13, 2019**")
    await channel.send("Discord Bot started on **December 16, 2019**")
    # Notify if verbose
    if debug == 1:
        await channel.send("Debug mode is enabled! Being verbose!")
    # Now on to business!
    #Declare vars
    storyContentStr = []
    storyNameStr = []
    # Alright, let's get the data from stories.txt
    i = 1
    f = open('stories.txt', 'r')
    for line in f.readlines():
        if i % 2 == 0 :
            storyContent = line
            storyContentStr.append(storyContent)	
        else:
            storyName = line
            storyNameStr.append(storyName)
        i+=1
    f.close()
    # Load files
    async with channel.typing():	
        #Count the number of lines in storyName, we use this to determine what story to use
        print(storyName)
        IntStoryCount = len(storyNameStr)
        await channel.send("Detected "+str(IntStoryCount)+" stories")
    # Randomly pick what story we will use
    story = random.randint(1, IntStoryCount)
    await channel.send(storyNameStr)
    # Print current story title, but remove the brackets first
    filteredTitle = re.findall(r'<(.*?)>', storyNameStr[story-1])
        
    # print the first result
    await channel.send("Current story title is  "+'"'+str(filteredTitle[0])+'"'+'\n')
# Alright, now onto the tricky part. We need to filter out all of the bracketed words in stories.txt, putting them into a list, replacing them with incremental strings. We also need to count how many there 	are for later.
# Pull all of the items with the <> brackets
    filtered = re.findall(r'<(.*?)>', storyContentStr[story-1])
    # We got them!
    if debug == 1:
        await channel.send(str(filtered))
    # Now we need to count them
    replacedNumber = len(filtered)
    
    # Run a loop to get the words
    
    replaceList = []
    #replaceList =['', '1', '2', '3', '4', '5', '6', '7', '8', '9', '10', '11', '12', '13', '14', '15', '16', '17', '18', '19', '20', '21', '22', '23', '24']
    replaceList.append("")
    await channel.send(str("Type a noun, verb, adjective, or adverb depending on what it asks you, followed by enter."))
    
    for loopCount in range(replacedNumber):
        print("Times looped: "+str(loopCount))
        #Wait for user to reply
        await channel.send("Give me a(n) "+"**"+str(filtered[loopCount])+"**"+": ")	
        # Push text to gTTS and save it to a file
        tts = gTTS(text="Give me a(n) "+str(filtered[loopCount])+": ")
        os.system("rm badCurrentTTS.mp3")
        os.system("rm currentTTS.wav")
        tts.save("badCurrentTTS.mp3")
       # gTTS is stupid and gives us a file that discord.py doesn't understand, so we have to convert it
        (
            ffmpeg
            .input('badCurrentTTS.mp3')
            .output('currentTTS.wav', audio_bitrate=48000, format='wav', sample_fmt='s16', ac='2')
            .run()
        )
        if voice.is_playing() == True:
            print("Audio is playing! Stopping playback!"+'\n')
            voice.stop()
        print("Attempting to play audio"+'\n')
        voice.play(discord.FFmpegPCMAudio("currentTTS.wav"))
        message = await client.wait_for('message', check=lambda message: message.channel.id == activeChannel)
        # , check=lambda messages: message.author.id == ctx.author.id and ctx.channel.id == message == ctx.message.id, timeout=30.0
        replaceVar = message.content
        print("You gave me: "+replaceVar)
        replaceList.append(replaceVar)
    print(replaceList)
# Run a loop to replace the words

    await channel.send("Replacing Words...")
    
    # Split the Story Content into a list
    storyContentList = re.split(r'<.*?>', storyContentStr[story-1])
    # Count the items in the list
    storyContentCount = len(storyContentList)
    x = 0
    for loopCount in range(storyContentCount):
        #print(storyContentList[loopCount])
        storyContentList.insert(x, replaceList[loopCount])
        x = x+2
    x = 0

    # Merge lists into a string
    generatedStory = ""
    generatedStory = generatedStory.join(storyContentList)
# Determine file name for file output
    now = datetime.now()
    currentDate = now.strftime("%d-%m-%Y-%H:%M:%S")
    saveFile = 'saved stories/generatedStory-'+currentDate
# Send Story to Discord
    await channel.send(generatedStory)
    await channel.send("Processing TTS, please wait!")
    async with channel.typing():
        tts = gTTS(text=generatedStory+"This story was generated by CCF_100's MadLibs.py", lang='en')
        os.system("rm badCurrentStory.mp3")
        tts.save("badCurrentStory.mp3")
        # gTTS is stupid and gives us a file that discord.py doesn't understand, so we have to convert it
        (
            ffmpeg
            .input('badCurrentStory.mp3')
            .output(saveFile+'.mp3', audio_bitrate=48000, format='wav', sample_fmt='s16', ac='2')
            .run()
        )
        if voice.is_playing() == True:
            print("Audio is playing! Stopping playback!"+'\n')
        voice.stop()
        print("Attempting to play audio"+'\n')
        voice.play(discord.FFmpegPCMAudio(saveFile+".mp3"))
    #exit()
    #Alright! We're done! Let's save the story to a file
    if os.path.exists("saved stories"):
        pass
    else:
        os.system("mkdir \"saved stories\"")    
    print("Saving story to .txt file")
    await channel.send("Saving story to .txt file")
    async with channel.typing():
        file = open(saveFile+'.txt', 'w+')	 
        line_offset = []
        offset = 0
        for line in file:
            line_offset.append(offset)
            offset += len(line)
        file.seek(0)
        file.write(filteredTitle[0]+'\n'+'\n')
        file.write(generatedStory)
        file.write('\n'+"Generated by Caleb Fontenot\'s madlibs.py")
        file.close()
    #Send generated .txt file to Discord
    await channel.send("Sending .txt file...")
    discordFile = discord.File(saveFile+'.txt', filename="generatedStory.txt")
    await channel.send(file=discordFile)
    #Send generated .mp3 file to Discord
    mp3File = saveFile+'.mp3'
    def file_size(fname):
        statinfo = os.stat(fname)
        return statinfo.st_size
    print("MP3 is "+str(file_size(mp3File))+" bytes.")
    await channel.send("Converted WAV is "+str(file_size(mp3File))+" bytes.")
    if int(file_size(mp3File)) <= int(8389999):
        # File is over 8 MiB! This will fail if we send the file corrected for transmission via Discord's voice chat. Let's send the original instead.
        discordFile = discord.File("badCurrentStory.mp3", filename=saveFile+'.mp3')
        await channel.send(file=discordFile)
    else:
        discordFile = discord.File(saveFile+'.wav', filename=saveFile+'.wav')
        await channel.send(file=discordFile)
        #await client.process_commands(message)
#Setup Discord functions and announce on discord that we are ready

class MyClient(discord.Client):  
    async def on_ready(self):
        global prefix
        #print(loop)
        print('Logged on as', self.user)
        channel = client.get_channel(656233549837631508)
        await channel.send("(Yet to be named) Discord bot, successfully connected!")
        await channel.send("Developed by CCF_100")
        if testing == 1:
            await channel.send("This bot is currently being worked on! Please don't start a game!")
        await channel.send("Run `mad!madlibs` to start madlibs")
        print("Ready!")		
        
    async def on_message(self, message, pass_context=True):
        if message.content == 'mad!madlibs':
            print("lol")
            channel = client.get_channel(656233549837631508)		
            await madlibsLoop()
            await channel.send("Done!")
"""
    async def on_message(self, message, pass_context=True):
        global prefix
        if message.content == prefix+'cough':
            voiceChannel = client.get_channel(690301271382556823)
            voice = await voiceChannel.connect()
            if voice.is_connected() == True:
                await voice.disconnect()
            voice = await voiceChannel.connect()
            voice.play(discord.FFmpegPCMAudio("coughing.wav"))
"""
    #Turn on message logging
""" 
    async def on_message(self, message, pass_context=True):
        if message.content == 'mad!logMessagesOn':
            pass
            if message.author.id == 294976590658666497:
                channel = client.get_channel(656233549837631508)
                await channel.send("Logging messages to console.")
                global logMessages
                logMessages = True
            else:
                channel = client.get_channel(656233549837631508)
                await channel.send("You are not authorized to use this command! Only @CCF_100#1050 may use this command!")
    #Turn off message logging
    async def on_message(self, message, pass_context=True):
        if message.content == 'mad!logMessagesOff':
            pass
            if message.author.id == 294976590658666497:
                channel = client.get_channel(656233549837631508)
                await channel.send("Stopping message logging!")
                global logMessages
                logMessages = False
            else:
                channel = client.get_channel(656233549837631508)
                await channel.send("You are not authorized to use this command! Only @CCF_100#1050 may use this command!")
"""
#Calls message listening function.
    #logMessages = False
    #async def messageListening():
        #global messageslist, messageAuthorList, logMessages
        #print("Now Listening for messages...")
        #messagesList= []
        #messageAuthorList = []
        #print("Loop start!")
        #x = 0
        #while True:
            #async def on_message(message):
                #raw_message = await client.wait_for('message')
                #if logMessages == True:
                    #print("Loop Count: "+str(x))
                #messagesList.append(raw_message.content)           
                #messageAuthorList.append(raw_message.author.nick)
                #if logMessages == True:
                    #print("Message from "+messageAuthorList[x]+": "+messagesList[x])
                #x += 1
    #Start messageListening in a new thread to prevent it from stalling the rest of the script
    #def startMessageListening():
        #asyncio.set_event_loop(loop)
        #loop.run_forever() new_loop = asyncio.new_event_loop()
    #t = Thread(target=startMessageListening, args=(new_loop,))
    #t.start()
  #Disconnect Voice
            #await asyncio.sleep(60)
            #voiceChannel = client.get_channel(682688245964079127)
            #await voice.disconnect()

#Run main loop
# The Discord bot ID isn't stored in this script for security reasons, so we have to go get it
f = open('botID.txt', 'r')
BotID = f.read()
#Cleanup from previous session
#os.system("rm badCurrentTTS.mp3")
#os.system("rm currentTTS.wav")
# Connect Bot To Discord and start running
client = MyClient()
client.run(BotID)
exit()


	# say the tts
	#print('\n'+"Processing Text-To-Speech, please wait..."+'\n')
	#tts = gTTS(text=generatedStory+"This story was generated by Caleb Fontenot's MadLibs.py", lang='en')
	#tts.save("TTS.mp3")
	##os.system("play TTS.mp3")
	#os.system("mv TTS.mp3 "+"\""+saveFile+".mp3"+"\"")
	#Start Discord Bot loop