Project Files Added

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+chromedriver filter=lfs diff=lfs merge=lfs -text

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2020 Vaibhav Singh
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

README.md

@@ -0,0 +1,20 @@
+# Omegle-Chatbot
+An Omegle Chatbot for promotion of Social media content or use it to increase views on YouTube. With the help of Chatterbot AI, this chatbot can be customized with new QnAs and will deal in a humanly way.    
+Used Selenium for Web Automation and English corpus data to train chatbot. [other languages](https://github.com/gunthercox/chatterbot-corpus/tree/master/chatterbot_corpus/data)
+    
+![](omg/saved2.png) 
+![](omg/saved4.png)
+## To setup and run
+`~$ python3 app.py`
+## ChatterBot Installation
+[If you are just getting started with ChatterBot](https://chatterbot.readthedocs.io/en/stable/setup.html)
+## To add custom dataset 
+[Setting the training class in chatt.py](https://chatterbot.readthedocs.io/en/stable/training.html)
+## Selenium
+[Getting started and setup](https://selenium-python.readthedocs.io/)
+## License & copyright
+© Vaibhav Singh     
+Licensed under the [MIT License](LICENSE).
+
+
+

app.py

@@ -0,0 +1,134 @@
+from selenium import webdriver 
+import time
+from time import sleep 
+from selenium.webdriver.common.keys import Keys
+
+
+class omegle():
+	def __init__(self, interests, chromedriver, headless = False):
+
+		self.driver = webdriver.Chrome(chromedriver) 
+		self.driver.get('https://www.omegle.com/')  
+		  
+		interest = self.driver.find_element_by_class_name('newtopicinput')
+
+		for i in interests:
+			interest.send_keys(i)
+			interest.send_keys(Keys.RETURN)
+
+		sleep(1)
+
+		self.driver.find_element_by_xpath('//*[@id="textbtn"]').click()
+
+		sleep(1)
+
+		try:
+			if self.driver.find_element_by_class_name("statuslog").text != None:
+				print('connected ....!')
+		except:
+			sleep(0.5)
+			pass
+		self.screen = 0
+
+	def getChat(self):
+		r = []
+
+		while True:
+			try:
+				texts = self.driver.find_elements_by_xpath("//p[@class='strangermsg']")
+				
+				for x in texts:
+					# print(x.text)
+					r.append(x.text)
+				break
+			except:
+				sleep(1)
+
+		return r
+
+	def sendChat(self, key = 'hello !'):
+		try:
+			send = self.driver.find_element_by_class_name('chatmsg')
+			send.send_keys(key)
+			send.send_keys(Keys.RETURN)
+		except:
+			print('fail')
+			pass
+		return
+
+	def newChat(self):
+		try:
+			dis = self.driver.find_element_by_xpath("//button[@class='disconnectbtn']")
+			dis.click()
+			dis.click()
+			dis.click()
+		except:
+			print('failed exiting !')
+
+	def checkChat(self):
+		try:
+			new = self.driver.find_element_by_class_name("newchatbtnwrapper")
+			dis = self.driver.find_element_by_xpath("//button[@class='disconnectbtn']")
+			self.driver.save_screenshot(f'./omg/screenie{self.screen}.png')
+			self.screen += 1
+			dis.click()
+			print('new chat !')
+			return True
+		except:
+			# print('failed new chat')
+			pass
+		return False
+
+	def reChat(self):
+		try:
+			self.driver.find_element_by_class_name("recaptcha-checkbox-border").click()
+		except:
+			print('failed recapta')
+
+interests = ['NY', 'italy', 'movies', 'justin beiber', 'netflix', 'money hiest', 'iron man', 'marvel','cap','india','mumbai']
+omg = omegle(interests = interests, chromedriver = './chromedriver')
+
+
+
+from chatt import Tron
+
+tron = Tron()
+	
+r,s = [0,0]
+o = 0
+while True:
+	
+	if r == 0 and s == 0:
+		s += 1
+		omg.sendChat()
+
+	sleep(1)
+
+	rl = omg.getChat()
+
+	if len(rl) > o:
+	
+		try:
+			text = rl[-1]
+			r+=1
+		except:
+			r = 0
+			s = 0
+			sleep(1)
+			continue
+		o = len(rl)
+
+		text = list(text.split(':'))
+		text = text[-1]
+		print("stranger : ",text)
+		st = tron.reply(text)
+		st = str(st)
+		print(st)
+		omg.sendChat(key = st)
+		s += 1
+
+	print(f"sent : {s}, recieved : {r}, lenr : {len(rl)}, o : {o}")
+	status = omg.checkChat()
+
+	if status:
+		s,r,o = 0,0,0

chatt.py

@@ -0,0 +1,40 @@
+from chatterbot import ChatBot
+from chatterbot.trainers import ChatterBotCorpusTrainer
+# from chatterbot.trainers import UbuntuCorpusTrainer
+
+# bot = ChatBot('Norman')
+class Tron():
+	def __init__(self):
+
+		self.bot = ChatBot(
+			'Norman',
+			storage_adapter='chatterbot.storage.SQLStorageAdapter',
+			logic_adapters=['chatterbot.logic.MathematicalEvaluation','chatterbot.logic.BestMatch'],
+			database_uri='sqlite:///database.sqlite3'
+		)
+
+		trainer = ChatterBotCorpusTrainer(self.bot)
+		trainer.train("chatterbot.corpus.english")
+		trainer.train("chatterbot.corpus.english.greetings")
+		trainer.train("chatterbot.corpus.english.conversations")
+
+	# trainer = UbuntuCorpusTrainer(bot)
+	# trainer.train()
+
+	def reply(self, msg):
+		if msg:
+			bot_out = self.bot.get_response(msg)
+			return bot_out
+
+
+	
+# agent = Tron()
+# while True:
+# 	inp = input("mark here.")
+# 	t = agent.reply(inp)
+# 	print(t,type(t))
+# 	s = str(t)
+# 	p = string(t)
+# 	print(s, type(s),type(p))
+
+	

chromedriver

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:722ae377fcf03cb8e26f65ae51ea5ded9257e299fc6e7f502e85eae9aad0f69d
+size 10251496