Display of the built poem.

app.py

@@ -25,48 +25,37 @@ def twolists(list1, list2):
     return newlist
 
 
-# This function structures the space already, set boundarys to the possible.
 def build_poem(source_lines, generated_lines):
+    # This function structures the space already, set boundarys to the possible.
     # return a list with the two lists interspersed
-    print("____build_poem")
-    print("____source_lines")
-    print(source_lines)
-    print("____generated_lines")
-    print(generated_lines)
-
     return twolists(source_lines, generated_lines)
 
 
 def add_to_lines(new_line):
-    print("____new_line")
-    print(new_line)
     source_lines.append(new_line)
-    # NOTE: pass the whole array of line to the generator
+    # NOTE: pass the whole array of lines to the generator
     # There is a compounding of the effect, a spiral of interaction.
     poem = build_poem(source_lines, generated_lines)
 
-    print("__poem")
-    print(poem)
-
-    # pass the full poem into the generator
-    result = generator("".join(poem), max_length=30, num_return_sequences=3)
-    # result = generator(poem, max_length=30, num_return_sequences=3)
-
-    print("____result")
-    print(result[0]["generated_text"])
-
+    # pass the last two lines of the poem
+    prompt = "\n".join(poem[-2:])
+    result = generator(prompt, max_length=50, num_return_sequences=3)
     response = result[0]["generated_text"]
 
-    # TODO: remove each source_line from the poem
-    response = response.replace("".join(poem), "")
-
-    print("____response")
-    print(response)
+    # Get only the new generated text
+    response = response.replace(prompt, "")
 
     generated_lines.append(response)
 
-    # html = [f"<p>{line}</p>" for line in build_poem(source_lines, generated_lines)]
-    html = [f"{line}<br/>" for line in build_poem(source_lines, generated_lines)]
+    lines = []
+    for line in build_poem(source_lines, generated_lines):
+        if (len(lines) % 2) == 1:
+            line = f"<p style='text-align: right;'>{line}</p>"
+        else:
+            line = f"<p>{line}</p>"
+        lines.append(line)
+
+    html = "".join(lines)
 
     return html