Yhhxhfh commited on
Commit
2d70365
1 Parent(s): 3c851d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -6
app.py CHANGED
@@ -23,29 +23,40 @@ from dotenv import load_dotenv
23
  from datetime import datetime
24
  from sklearn.feature_extraction.text import CountVectorizer
25
  from sklearn.model_selection import train_test_split
 
 
 
26
 
27
  load_dotenv()
28
 
29
  app = FastAPI()
30
 
31
  lemmatizer = WordNetLemmatizer()
32
- redis_password = os.getenv("REDIS_PASSWORD")
33
- r = redis.Redis(host=os.getenv("REDIS_HOST"), port=int(os.getenv("REDIS_PORT")), password=redis_password)
 
 
34
 
35
  def create_intents_json():
36
  intents = {
37
  "intents": [
38
  {
39
  "tag": "greeting",
40
- "patterns": ["Hola", "¿Cómo estás?", "Buenos días"],
41
  "responses": ["¡Hola!", "¿Cómo puedo ayudarte?"],
42
- "date": "2021-01-01"
43
  },
44
  {
45
  "tag": "goodbye",
46
- "patterns": ["Adiós", "Hasta luego", "Nos vemos"],
47
  "responses": ["¡Hasta luego!", "Cuídate!"],
48
- "date": "2021-01-01"
 
 
 
 
 
 
49
  }
50
  ]
51
  }
 
23
  from datetime import datetime
24
  from sklearn.feature_extraction.text import CountVectorizer
25
  from sklearn.model_selection import train_test_split
26
+ from faker import Faker
27
+ from random_word import RandomWords
28
+ from textgenrnn import textgenrnn
29
 
30
  load_dotenv()
31
 
32
  app = FastAPI()
33
 
34
  lemmatizer = WordNetLemmatizer()
35
+ faker = Faker()
36
+ r = redis.Redis(host=os.getenv("REDIS_HOST"), port=int(os.getenv("REDIS_PORT")), password=os.getenv("REDIS_PASSWORD"))
37
+ random_words = RandomWords()
38
+ textgen = textgenrnn()
39
 
40
  def create_intents_json():
41
  intents = {
42
  "intents": [
43
  {
44
  "tag": "greeting",
45
+ "patterns": [faker.sentence() for _ in range(5)],
46
  "responses": ["¡Hola!", "¿Cómo puedo ayudarte?"],
47
+ "date": datetime.now().strftime("%Y-%m-%d")
48
  },
49
  {
50
  "tag": "goodbye",
51
+ "patterns": [faker.sentence() for _ in range(5)],
52
  "responses": ["¡Hasta luego!", "Cuídate!"],
53
+ "date": datetime.now().strftime("%Y-%m-%d")
54
+ },
55
+ {
56
+ "tag": "random_word",
57
+ "patterns": [random_words.get_random_word() for _ in range(5)],
58
+ "responses": [faker.sentence() for _ in range(5)],
59
+ "date": datetime.now().strftime("%Y-%m-%d")
60
  }
61
  ]
62
  }