Update app.py

app.py

@@ -1269,31 +1269,31 @@ def translate(txt,to_lang="en",from_lang="auto"):
     log(f'RET translate with translation as {translation}')
     return translation.lower()
 
-@spaces.GPU(duration=120)
+@spaces.GPU(duration=240)
 def handle_generation(artist,song,genre,lyrics):
 
     log(f'CALL handle_generate')
 
     pos_artist = re.sub(r"([ \t]){1,}", " ", artist).upper().strip()
+    
     pos_song = re.sub(r"([ \t]){1,}", " ", song).lower().strip()
     pos_song = ' '.join(word[0].upper() + word[1:] for word in pos_song.split())
 
-    lyrics = re.sub(r"[,]",".",lyrics)
+    lyrics = re.sub(r",( ){1,}",". ",lyrics)
     lyrics_lines = re.split(r"([\n]){1,}", lyrics)
     
     for line_index in range(len(lyrics_lines)):
-        if re.sub(f'[{punctuation}]$', '', lyrics_lines[line_index]) == lyrics_lines[line_index]:
+        lyrics_lines[line_index] = lyrics_lines[line_index].strip()
+        if re.sub(r'[\.]$', '', lyrics_lines[line_index]) == lyrics_lines[line_index]:
             lyrics_lines[line_index] = lyrics_lines[line_index].strip() + "."
-        else:
-            lyrics_lines[line_index] = lyrics_lines[line_index].strip()
     lyrics = " ".join(lyrics_lines)
     
-    pos_genre = re.sub(f'[{punctuation}]', '', re.sub(r"([ \t]){1,}", " ", genre)).lower().strip()
+    pos_genre = re.sub(r"([ \t]){1,}", " ", genre).lower().strip()
     pos_genre = ' '.join(word[0].upper() + word[1:] for word in pos_genre.split())
 
     pos_lyrics = re.sub(r"([ \t]){1,}", " ", lyrics).lower().strip()
     pos_lyrics = pos_lyrics if pos_lyrics == "" else summarize(translate(pos_lyrics))
-    pos_lyrics = re.sub(r"([ \t]){1,}", " ", re.sub(f'[{punctuation}]', '', pos_lyrics)).lower().strip()
+    pos_lyrics = re.sub(r"([ \t]){1,}", " ", pos_lyrics).lower().strip()
 
     neg = f"Sexuality, Human body, Human, Textual, Labeled, Distorted, Discontinuous, Blurry, Doll-Like, Overly Plastic, Low-Quality, Painted, Smooth, Artificial, Phony, Gaudy, Digital Effects."
     q = "\""