maybe better fallback?

app.py

@@ -80,7 +80,74 @@ def npc_randomize():
 
     
 
-
+def genericDescription():
+    colors="""red
+    blue
+    green
+    yellow
+    orange
+    purple
+    pink
+    brown
+    black
+    white""".split("\n")
+    colors=[x.strip() for x in colors]
+    outfits="""shirt
+    pair of pants
+    pair of shoes
+    hat
+    pair of glasses
+    backpack
+    belt
+    tie
+    cloak
+    robe
+    chain mail vest
+    suit of plate armor
+    suit of leather armor
+    suit of studded leather armor
+    suit of scale armor
+    suit of chain mail armor
+    suit of ring mail armor
+    """.split("\n")
+    outfits=[x.strip() for x in outfits]
+    weapons="""sword
+    dagger
+    mace
+    axe
+    polearm
+    bow
+    crossbow
+    sling
+    club
+    flail
+    warhammer
+    morningstar
+    halberd
+    war pick
+    war sickle
+    war hammer""".split("\n")
+    weapons=[x.strip() for x in weapons]
+
+    objects="""shield
+    lantern
+    sack
+    severed head
+    crystal""".split("\n")
+    objects=[x.strip() for x in objects]
+    
+    desc="wears a {color} {outfit}".format(color=random.choice(colors),outfit=random.choice(outfits))
+    if random.random()<0.5:
+        desc+="and a {color} {outfit}".format(color=random.choice(colors),outfit=random.choice(outfits))
+    
+    if random.random()<0.5:
+        desc+="and carries a {weapon}".format(weapon=random.choice(weapons))
+    elif random.random()<0.5:
+        desc+="and carries a {weapon} and a {object}".format(weapon=random.choice(weapons),object=random.choice(objects))
+    else:
+        desc+="and carries two {weapon}s".format(weapon=random.choice(weapons))
+        
+    return desc
 
 
 def npc_generate(name,race,characterClass,pronoun):
@@ -88,7 +155,7 @@ def npc_generate(name,race,characterClass,pronoun):
   desc="{name} is a {race} {characterClass}, {pronoun}".format(name=name,race=race,characterClass=characterClass,pronoun=pronoun)
 
   p = prompt + "\n"+desc
-  print(f"*****Inside poem_generate - Prompt is :{p}")
+  print(f"*****Inside desc_generate - Prompt is :{p}")
   json_ = {"inputs": p,
             "parameters":
             {
@@ -101,35 +168,41 @@ def npc_generate(name,race,characterClass,pronoun):
   response = requests.post(API_URL, json=json_)
   output = response.json()
   print(f"If there was an error? Reason is : {output}")
-  output_tmp = output[0]['generated_text']
-  print(f"GPTJ response without splits is: {output_tmp}")
-  #poem = output[0]['generated_text'].split("\n\n")[0] # +"."
-  if "\n\n" not in output_tmp:
-    if output_tmp.find('.') != -1:
-      idx = output_tmp.find('.')
-      poem = output_tmp[:idx+1]
-    else:
-      idx = output_tmp.rfind('\n')
-      poem = output_tmp[:idx]
+
+
+  #error handling
+  if "error" in output:
+    #fallback method
+    longDescription=genericDescription()
   else:
-    poem = output_tmp.split("\n\n")[0] # +"."
-  poem = poem.replace('?','')
-  print(f"Poem being returned is: {poem}")
-  return desc+poem
-
-def poem_to_image(poem):
-  print("*****Inside Poem_to_image")
-  poem = " ".join(poem.split('\n'))
-  poem = poem + ", character art, concept art, artstation"
+    output_tmp = output[0]['generated_text']
+    print(f"GPTJ response without splits is: {output_tmp}")
+    if "\n\n" not in output_tmp:
+        if output_tmp.find('.') != -1:
+        idx = output_tmp.find('.')
+        longDescription = output_tmp[:idx+1]
+        else:
+        idx = output_tmp.rfind('\n')
+        longDescription = output_tmp[:idx]
+    else:
+        longDescription = output_tmp.split("\n\n")[0] # +"."
+    longDescription = longDescription.replace('?','')
+    print(f"longDescription being returned is: {longDescription}")
+  return desc+longDescription
+
+def desc_to_image(desc):
+  print("*****Inside desc_to_image")
+  desc = " ".join(desc.split('\n'))
+  desc = desc + ", character art, concept art, artstation"
   steps, width, height, images, diversity = '50','256','256','1',15
   iface = gr.Interface.load("spaces/multimodalart/latentdiffusion")
   print("about to die",iface,dir(iface))
 
-  prompt = re.sub(r'[^a-zA-Z0-9 ,.]', '', poem)
+  prompt = re.sub(r'[^a-zA-Z0-9 ,.]', '', desc)
   print("about to die",prompt)
 
 
-  img=iface(poem, steps, width, height, images, diversity)[0]
+  img=iface(desc, steps, width, height, images, diversity)[0]
   return img
 
 demo = gr.Blocks()
@@ -155,12 +228,12 @@ with demo:
     input_pronoun = gr.Textbox(label="pronoun",placeholder="he")
 
   with gr.Row():
-    poem_txt = gr.Textbox(label="description",lines=7)
+    desc_txt = gr.Textbox(label="description",lines=7)
     output_image = gr.Image(label="portrait",type="filepath", shape=(256,256))
   
   b0.click(npc_randomize,inputs=[],outputs=[input_name,input_race,input_class,input_pronoun])
-  b1.click(npc_generate, inputs=[ input_name,input_race,input_class,input_pronoun], outputs=poem_txt)
-  b2.click(poem_to_image, poem_txt, output_image)
+  b1.click(npc_generate, inputs=[ input_name,input_race,input_class,input_pronoun], outputs=desc_txt)
+  b2.click(desc_to_image, desc_txt, output_image)
   #examples=examples
 
 demo.launch(enable_queue=True, debug=True)