placed the whole interface more into blocks instead of chatinterface

app.py

@@ -210,11 +210,24 @@ def bot_comms(message, history):
                     buffer += text
                 yield "".join(gpt_outputs)
 
-chatbot = gr.Chatbot(height=600, label="Chimera AI")
-chat_input = gr.MultimodalTextbox(interactive=True, file_types=["images"], placeholder="Enter your question or upload an image.", show_label=False)
 
 with gr.Blocks(fill_height=True) as demo:
     gr.Markdown(DESCRIPTION)
+    
+    chatbot = gr.Chatbot(height=600, label="Chimera AI")
+    chat_input = gr.MultimodalTextbox(interactive=True, file_types=["images"], placeholder="Enter your question or upload an image.", show_label=False)
+    image_output = gr.Image(type="pil", label="Generated Image")
+    
+    def process_response(message, history):
+        response = bot_comms(message, history)
+        if isinstance(response, tuple) and len(response) == 2:
+            text, image = response
+            return text, image
+        return response, None
+
+    chatbot_output = gr.Chatbot(height=600, label="Chimera AI")
+
+    chat_input.submit(process_response, inputs=[chat_input, chatbot], outputs=[chat_output, image_output])
     # if mode_gradio == "imagery":
     #     gr.Interface()
     # Customize chatinterface to handle tuples
@@ -228,13 +241,13 @@ with gr.Blocks(fill_height=True) as demo:
     #         else:
     #             output.append(item)
     #     return output
-    gr.ChatInterface(
-        fn=bot_comms,
-        chatbot=chatbot,
-        fill_height=True,
-        multimodal=True,
-        textbox=chat_input,
-    )
+    # gr.ChatInterface(
+    #     fn=bot_comms,
+    #     chatbot=chatbot,
+    #     fill_height=True,
+    #     multimodal=True,
+    #     textbox=chat_input,
+    # )
 
 if __name__ == "__main__":
     demo.launch()