Update app.py

app.py

@@ -45,6 +45,7 @@ def transcribe(audio,prompt_num,user_keywords):
 
 def keywords(text,prompt_num,user_keywords):
 
+    transcription = text
 
     kw_model = KeyBERT()
     a = kw_model.extract_keywords(text, keyphrase_ngram_range=(1, 3), stop_words=None)
@@ -134,10 +135,10 @@ def keywords(text,prompt_num,user_keywords):
     imgs_comb = np.hstack([i.resize(min_shape) for i in images])
     imgs_comb = Image.fromarray( imgs_comb)
     
-    return images
+    return images,transcription,keyword_pool,generated_prompts
 
 speech_text = gr.Interface(fn=transcribe, inputs=[gr.Audio(source="microphone", type="filepath"),gr.Number(placeholder = "Number of Images to be generated (int): "),gr.Textbox(placeholder = "Additional keywords (comma delimitied): ")], outputs=["text","number","text"], title = 'Speech to Image Generator', enable_queue=True)
-text_prompts = gr.Interface(fn=keywords, inputs=["text","number","text"], outputs=gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[2], height="auto"), title = 'Speech to Image Generator', enable_queue=True)
+text_prompts = gr.Interface(fn=keywords, inputs=["text","number","text"], outputs=[gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[2], height="auto"),"text","text","text"], title = 'Speech to Image Generator', enable_queue=True)
 
 gr.Series(speech_text,text_prompts).launch(inline = False, enable_queue=True).queue()