Update app.py

app.py

@@ -27,25 +27,12 @@ def generate_music(Prompt):
 
 def process_input(Prompt,choice):
     if choice == "Chat":
-        return vanilla_chatbot(text),None,None
+        return vanilla_chatbot(Prompt),None,None
     elif choice == 'Music':
         rate,audio = generate_music(Prompt)
         return None, (rate,audio), None
     else:
         return None , None , generate_image(Prompt)
 
-with gr.Blocks as demo:
-    with gr.Row():
-        text_input = gr.Textbox()
-        choice = gr.Radio(choices=["Chat","Music","Image"])
-
-    with gr.Row():
-        chatbot_output = gr.Textbox()
-        music_output =gr.Audio()
-        image_output =gr.Image()
-        
-    submit_btn = gr.Button("Generate")
-    
-    submit_btn.click(fn=process_input,inputs=[text_input,choice],outputs=[chatbot_output,music_output,image_output])
-
+demo=gr.Interface(fn=process_input,inputs=["text",gr.Radio(["chat","music","image"])],outputs=["text", "audio","image"],title="Multimodal assistance")
 demo.launch(debug=True)