Update app.py

app.py

@@ -66,6 +66,7 @@ def inference(audio, prompt, model, temperature, latest):
     infers = []
     temp = []
     #infered=[]
+    
     for i in range(5):
         print("print1 ", response['choices'][i]['text'])
         temp.append(response['choices'][i]['text'])
@@ -76,14 +77,26 @@ def inference(audio, prompt, model, temperature, latest):
         #print("Infered type is: ", type(infered))
         infers = list(map(lambda x: x.replace("\n", ""), temp))
         #infered = list(map(lambda x: x.split(','), infers))
+
+    op1 = infers[0]
+    op2 = infers[1]
+    op3 = infers[2]
+    op4 = infers[3]
+    op5 = infers[4]
+    
+    
     print("INFERS TYPE: ", type(infers))
 
     convoState = latest
-    infersStr = str(infers)
+    #infersStr = str(infers)
         
 
-    return transcript, infersStr, convoState
-
+    return transcript, op1, op2, op3, op4, op5, convoState
+    
+def appendPrediction(val, convoState):
+    convoState.append(val)
+    return convoState
+    
 # get audio from microphone 
 with gr.Blocks() as face:
     
@@ -97,7 +110,12 @@ with gr.Blocks() as face:
             transcribe_btn = gr.Button(value="Transcribe")
         with gr.Column():
             script = gr.Textbox(label="Transcribed text")
-            options = gr.Textbox(label="Predictions")
+            #options = gr.Textbox(label="Predictions")
+            option1 = gr.Button(value="  ")
+            option2 = gr.Button(value="  ")
+            option3 = gr.Button(value="  ")
+            option4 = gr.Button(value="  ")
+            option5 = gr.Button(value="  ")
             #options = gr.Dataset(components=[gr.Radio], samples=["One", "Two", "Three", "Four", "Five"])
             '''options = gr.Dataset(components=[gr.Textbox(visible=False)],
                 label="Text Dataset",
@@ -112,7 +130,7 @@ with gr.Blocks() as face:
             #options = gr.Radio(choices=["One", "Two", "Three", "Four", "Five"])
             latestConvo = gr.Textbox(label="Running conversation")
             #transcribe_btn.click(inference)
-    transcribe_btn.click(fn=inference, inputs=[audio, promptText, dropChoice, sliderChoice, convoState], outputs=[script, options, latestConvo])
+    transcribe_btn.click(fn=inference, inputs=[audio, promptText, dropChoice, sliderChoice, convoState], outputs=[script, option1, option2, option3, option4, option5, latestConvo])
     #examples = gr.Examples(examples=["Sedan, Truck, SUV", "Dalmaion, Shepherd, Lab, Mutt"], inputs=[options])