Update app.py

app.py

@@ -83,9 +83,9 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
     for response in stream:
         output += response.token.text
         if history:
-            yield "",[(prompt,output)],stats,None
+            yield "",[(prompt,output)],stats,None,None
         else:
-            yield "",[(prompt,output)],stats,None
+            yield "",[(prompt,output)],stats,None,None
     generate_kwargs2 = dict(
         temperature=temperature,
         max_new_tokens=128,
@@ -105,6 +105,7 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
     lines = output.strip().strip("\n").split("\n")
     skills=[]
     skill_dict={}
+    option_drop=[]
     new_stat="*******************\n"
     for i,line in enumerate(lines):
         if ":" in line:
@@ -112,6 +113,7 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
                 try:
                     if f'{z}' in lines[i+z]:
                         print(lines[i+z].split(" ",1)[1])
+                        option_drop.append(lines[i+z].split(" ",1)[1])
                 except Exception:
                     pass
         if ": " in line:
@@ -132,11 +134,13 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
     skills.append(skill_dict)
     new_stat+="*******************\n"
     stats=new_stat
+    option_drop=gr.Dropdown(label="Choices", choices=[e for e in option_drop])
+
     if history:
         history.append((prompt,output))
-        yield "",history,stats,skills
+        yield "",history,stats,skills,option_drop
     else:
-        yield "",[(prompt,output)],stats,skills
+        yield "",[(prompt,output)],stats,skills,option_drop
 
 def clear_fn():
     return None,None
@@ -160,9 +164,9 @@ with gr.Blocks() as app:
                 with gr.Row():
                     with gr.Column(scale=3):
                         prompt=gr.Textbox(label = "Prompt", value="Start a new game")
-                    with gr.Column(scale=1):
+                    with gr.Column(scale=2):
                         button=gr.Button()
-                    
+                        opt=gr.Dropdown(label="Choices")
                     #models_dd=gr.Dropdown(choices=[m for m in return_list],interactive=True)
                 with gr.Row():
                     stop_button=gr.Button("Stop")
@@ -172,11 +176,15 @@ with gr.Blocks() as app:
             with gr.Column(scale=1):
                 json_out=gr.JSON(value=base_stats)
                 char_stats=gr.Textbox(value=text_stats)
+                textboxes = []
+                for i in range(max_textboxes):
+                    t = gr.Button(f"Textbox {i}")
+                textboxes.append(t)
     #text=gr.JSON()
     #inp_query.change(search_models,inp_query,models_dd)
     #test_b=test_btn.click(itt,url,e_box)
     clear_btn.click(clear_fn,None,[prompt,chatbot])
-    go=button.click(generate,[prompt,chatbot,tokens,char_stats],[prompt,chatbot,char_stats,json_out])
+    go=button.click(generate,[prompt,chatbot,tokens,char_stats],[prompt,chatbot,char_stats,json_out,opt])
     stop_button.click(None,None,None,cancels=[go])
 app.launch(show_api=False)