examples

app.py

@@ -2,24 +2,26 @@ import gradio as gr
 import random
 import markovify
 
-def gen(text):
+def gen(text, size):
   text = text.lower()
-  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(100) or random.choice(
-      db.splitlines())[:100] + random.choice(symbolsplus)
+  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)
 
   return sentence
 
 iface = gr.Interface(
     fn=gen,
-    inputs="text",
+    inputs="text", "size",
     outputs="text",
     examples=[["Привет!"], ["Как дела?"]])
 iface.launch()