Update app.py

app.py

@@ -49,7 +49,7 @@ def get_rhymes(inp, level):
 def get_inputs_length(input):
     input_ids = gpt2_tokenizer(input)['input_ids']
     return len(input_ids)
-    
+	
 def get_prediction(sent):
     
     token_ids = roberta_tokenizer.encode(sent, return_tensors='pt')
@@ -79,13 +79,16 @@ def get_prediction(sent):
         best_guess = best_guess+" "+j[0]
         
     return best_guess
-
+	
 def get_line(prompt, inputs_len):
     line = gpt2_model.generate_one(prompt=prompt + ".", max_length=inputs_len + 7, min_length=4)[len(prompt)+2:]
     return line
 
 def get_rhyming_line(prompt, rhyming_word, inputs_len):
     gpt2_sentence = gpt2_model.generate_one(prompt=prompt + ".", max_length=inputs_len + 4, min_length=2)[len(prompt)+2:]
+    while len(gpt2_sentence) == 0:
+        gpt2_sentence = gpt2_model.generate_one(prompt=prompt + ".", max_length=inputs_len + 4, min_length=2)[len(prompt)+2:]
+    
     gpt2_sentence = gpt2_sentence.replace("\n", "")
     print(f"\nGetting rhyming line starting with '{gpt2_sentence}' and ending with rhyming word '{rhyming_word}'")
     sentence = gpt2_sentence + " ___ ___ ___ " + rhyming_word
@@ -104,11 +107,14 @@ def get_rhyming_line(prompt, rhyming_word, inputs_len):
     return final_sentence
 
 def gpt2_summary(topic):
-    return gpt2_model.generate_one(prompt=f"Here is some information about {topic}", top_k=100, top_p=0.95)
-
+    return gpt2_model.generate_one(prompt=f"Here is some information about {topic}.", top_k=100, top_p=0.95, min_length=200)
+	
 def generate(topic, wiki=True):
     if wiki:
-        topic_summary = remove_punctuation(wikipedia.summary(topic))
+        try:
+            topic_summary = remove_punctuation(wikipedia.summary(topic))
+        except:
+            return(f"Method A struggled to find information about {topic}, please try a different topic!")
     else:
         topic_summary = remove_punctuation(gpt2_summary(topic))
 
@@ -184,25 +190,44 @@ def generate(topic, wiki=True):
     print(limerick)
 
     return limerick
-
+	
 def compare_summaries(topic):
     wiki_limerick = generate(topic)
     gpt2_limerick = generate(topic, wiki=False)
 
-    output1 = f"Limerick with Wikipedia summary of topic as prompt: \n"
-    output1 += wiki_limerick + "\n"
-    output2 = f"Limerick with GPT-2 summary of topic as prompt: \n"
-    output2 += gpt2_limerick
+    output1 = wiki_limerick
+    output2 = gpt2_limerick
+    print(output1 + "\n" + output2)
 
     return output1, output2
-
+	
 import gradio as gr
+description = "Generates limericks (five-line poems with a rhyme scheme of AABBA) via two different methods"
+article = '<center><big><strong>Limerick Generation</strong></big></center>'\
+        '<center>By Ans Farooq</center>'\
+        '<strong>Description</strong><br>'\
+        'Recent advances in natural language processing (NLP) have shown '\
+        'incredible promise at generating human-quality language. Poetry '\
+        'presents an additional challenge as it often relies on rhyme and '\
+        'rhythm of language. Factoring these in presents an interesting '\
+        'challenge to new deep learning-based methods. This text-generation '\
+        'project examines the use of transformer-based deep learning methods '\
+        'and the addition of constraints for length, rhyme and rhythm given '\
+        'example words to seed a poem. This interface allows you to produce two '\
+        'cohesive limericks automatically, using two different methods. The '\
+        'results of this project are to be evaluated through human comparisons.'
+
+input = gr.inputs.Textbox(label='Topic')
+output1 = gr.outputs.Textbox(label='Method A')
+output2 = gr.outputs.Textbox(label='Method B')
 
 interface = gr.Interface(
     fn=compare_summaries, 
-    inputs="text", 
-    outputs=["text", "text"],
+    inputs=input, 
+    outputs=[output1, output2],
     title="Text-generation with rhyme and rhythm",
     layout="horizontal",
-    theme="peach")
+    theme="peach",
+    description=description,
+    article=article)
 interface.launch(debug=True)