add algo to generate one song

README.md

@@ -39,4 +39,105 @@ Ich hab' Jungs in der Trap, ich hab' Jungs an der Uni (Ahh)
 [Post-Hook_nullsechsroy_Goodies]
 Ja, ich weiß, sie findet niemals ein'n wie mich (Ahh)
 ...
+```
+
+# Source code to create a song
+
+```
+from transformers import pipeline, AutoTokenizer,AutoModelForCausalLM
+
+# load the model from huggingface
+rap_model = AutoModelForCausalLM.from_pretrained("Bachstelze/poetryRapGPT")
+tokenizer = AutoTokenizer.from_pretrained("Anjoe/german-poetry-gpt2")
+rap_pipe = pipeline('text-generation',
+                    model=rap_model,
+                    tokenizer=german_gpt_model,
+                    pad_token_id=tokenizer.eos_token_id,
+                    max_length=250)
+                    
+# set the artist 
+song_artist = "nullsechsroy" # "nullsechsroy Deluxe"
+# add a title idea or leave it blank
+title = "" # "Kristall" "Fit"
+
+# definition of the song structure
+type_with_linenumbers = [("Intro",4),
+                        ("Hook",4),
+                        ("Part 1",6),
+                        ("Part 2",6),
+                        ("Outro",4)]
+                
+
+def set_title(song_parts):
+  """
+  we create a title if it isn't set already
+  and add the title to the songs parts dictionary
+  """
+  if len(title) > 0:
+    song_parts["Title"] = "\n[Title_" + song_artist + "_" + title + "]\n"
+    song_parts["artist_with_title"] = song_artist + "_" + title
+  else:
+    title_input = "\n[Title_" + song_artist + "_"
+    title_lines = rap_pipe(title_input)[0]['generated_text']
+    index_title_end = title_lines.index("]\n")
+    artist_with_title = title_lines[8:index_title_end]
+    song_parts["Title"] =  title_lines[:index_title_end+1]
+    song_parts["artist_with_title"] = artist_with_title
+  
+def create_song_by_parts():
+  """
+  we iterate over the song structure
+  and return the dictionary with the song parts
+  """
+  song_parts = {}
+  set_title(song_parts)
+
+  for (part_type, line_number) in type_with_linenumbers:
+    new_song_part = create_song_part(part_type, song_parts["artist_with_title"], line_number)
+    song_parts[part_type] = new_song_part
+
+  return song_parts
+
+def get_line(pipe_input, line_number):
+  """
+  We generate a new song line.
+  This function could be scaled to more lines.
+  """
+  new_lines = rap_pipe(pipe_input)[0]['generated_text'].split("\n")
+  if len(new_lines) > line_number + 3:
+    new_line = new_lines[line_number+3] + "\n"
+    return new_line
+  else: #retry
+    return get_line(pipe_input, line_number)
+
+def create_song_part(part_type, artist_with_title, lines_number):
+  """
+  we generate one song part
+  """
+  start_type = "\n["+part_type+"_"+artist_with_title+"]\n"
+  song_part = start_type # + preset start line
+  lines = [""]
+  
+  for line_number in range(lines_number):
+    pipe_input = start_type + lines[-1]
+    new_line = get_line(pipe_input, line_number)
+    lines.append(new_line)
+    song_part += new_line
+  return song_part
+
+def print_song(song_parts):
+  """
+  Let's print the generated song
+  """
+  print(song_parts["Title"])
+  print(song_parts["Intro"])
+  print(song_parts["Part 1"])
+  print(song_parts["Hook"])
+  print(song_parts["Part 2"])
+  print(song_parts["Hook"])
+  print(song_parts["Outro"])
+
+# start the generation of one song
+song_parts = create_song_by_parts()
+print_song(song_parts)
 ```