Logan Zoellner commited on
Commit
e44be4d
·
1 Parent(s): 63acdcd

fix bug and truncate story

Browse files
Files changed (1) hide show
  1. app.py +10 -3
app.py CHANGED
@@ -53,6 +53,15 @@ def continue_story(prompt):
53
  #check if response starts with "computer:", if not add it
54
  if ":" not in output_tmp:
55
  output_tmp = "computer:"+output_tmp
 
 
 
 
 
 
 
 
 
56
  story=story+"player:"+prompt+"\n"+output_tmp+"\n"
57
 
58
 
@@ -76,9 +85,7 @@ with demo:
76
  with gr.Row():
77
  story = gr.Textbox(label="description",lines=7)
78
 
79
- b0.click(continue_story,inputs=[],outputs=[input_name,input_race,input_class,input_pronoun])
80
- b1.click(npc_generate, inputs=[ input_name,input_race,input_class,input_pronoun], outputs=desc_txt)
81
- b2.click(desc_to_image, desc_txt, output_image)
82
  #examples=examples
83
 
84
  demo.launch(enable_queue=True, debug=True)
 
53
  #check if response starts with "computer:", if not add it
54
  if ":" not in output_tmp:
55
  output_tmp = "computer:"+output_tmp
56
+
57
+
58
+
59
+ #truncate story to last 6 lines
60
+ story_tmp = story.split("\n")
61
+ if len(story_tmp)>6:
62
+ story_tmp = story_tmp[-6:]
63
+ story = "\n".join(story_tmp)
64
+ #return story
65
  story=story+"player:"+prompt+"\n"+output_tmp+"\n"
66
 
67
 
 
85
  with gr.Row():
86
  story = gr.Textbox(label="description",lines=7)
87
 
88
+ b0.click(continue_story,inputs=[input_command],outputs=[story])
 
 
89
  #examples=examples
90
 
91
  demo.launch(enable_queue=True, debug=True)