examples

app.py

@@ -2,26 +2,25 @@ import gradio as gr
 import random
 import markovify
 
-def gen(text, size):
+def gen(text):
   text = text.lower()
-  size = int(size)
-  if size.isnumeric():
-    file_name = "./dataset.txt"
-    with open(file_name, "a", encoding="utf-8") as f:
-      f.write(f"{text}\n")
-    with open(file_name, encoding="utf-8") as f:
-      db = f.read()
-    db = db.strip().lower()
-    text_model = markovify.NewlineText(input_text=db, state_size=1, well_formed=False)
-    symbolsplus = [".","~~","!","?"]
-    sentence: str = text_model.make_short_sentence(size) or random.choice(
-        db.splitlines())[:size] + random.choice(symbolsplus)
+  size = random.randint(1, 650)
+  file_name = "./dataset.txt"
+  with open(file_name, "a", encoding="utf-8") as f:
+    f.write(f"{text}\n")
+  with open(file_name, encoding="utf-8") as f:
+    db = f.read()
+  db = db.strip().lower()
+  text_model = markovify.NewlineText(input_text=db, state_size=1, well_formed=False)
+  symbolsplus = [".","~~","!","?"]
+  sentence: str = text_model.make_short_sentence(size) or random.choice(
+      db.splitlines())[:size] + random.choice(symbolsplus)
 
   return sentence
 
 iface = gr.Interface(
     fn=gen,
-    inputs=["text", "size"],
+    inputs="text",
     outputs="text",
     examples=[["Привет!"], ["Как дела?"]])
 iface.launch()