Update app.py

app.py

@@ -8,6 +8,7 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 from transformers import pipeline
 from datasets import load_dataset
+import time
 
 dataset = load_dataset("ibunescu/qa_legal_dataset_train")
 
@@ -182,7 +183,23 @@ def chat_between_bots(system_message1, system_message2, max_tokens, temperature,
 def update_pdf_gallery(pdf_files):
     return pdf_files
 
-# Gradio interface
+# Function to add message to the chatbot
+def add_message(history, message):
+    for x in message["files"]:
+        history.append(((x,), None))
+    if message["text"] is not None:
+        history.append((message["text"], None))
+    return history, gr.MultimodalTextbox(value=None, interactive=False)
+
+# Bot function to simulate response
+def bot(history):
+    response = "**That's cool!**"
+    history[-1][1] = ""
+    for character in response:
+        history[-1][1] += character
+        time.sleep(0.05)
+        yield history
+
 with gr.Blocks(css=custom_css) as demo:
     history1 = gr.State([])
     history2 = gr.State([])
@@ -208,7 +225,7 @@ with gr.Blocks(css=custom_css) as demo:
             defense_score_color = gr.HTML()
     
     shared_argument = gr.Textbox(label="Case Outcome", interactive=True)
-    pdf_upload = gr.File(label="Upload Case Files (PDF)", file_types=[".pdf"], multiple=True)
+    pdf_upload = gr.File(label="Upload Case Files (PDF)", file_types=[".pdf"])
     pdf_gallery = gr.Gallery(label="PDF Gallery")
     
     submit_btn = gr.Button("Argue")
@@ -217,5 +234,20 @@ with gr.Blocks(css=custom_css) as demo:
     submit_btn.click(chat_between_bots, inputs=[system_message1, system_message2, max_tokens, temperature, top_p, history1, history2, shared_history, message], outputs=[prosecutor_response, defense_response, history1, history2, shared_history, shared_argument, prosecutor_score_color, defense_score_color])
     pdf_upload_btn.click(update_pdf_gallery, inputs=[pdf_upload], outputs=[pdf_gallery, pdf_files])
     
-if __name__ == "__main__":
-    demo.launch()
+    # Chatbot section
+    chatbot = gr.Chatbot(
+        [],
+        elem_id="chatbot",
+        bubble_full_width=False
+    )
+
+    chat_input = gr.MultimodalTextbox(interactive=True, file_types=["image"], placeholder="Enter message or upload file...", show_label=False)
+
+    chat_msg = chat_input.submit(add_message, [chatbot, chat_input], [chatbot, chat_input])
+    bot_msg = chat_msg.then(bot, chatbot, chatbot, api_name="bot_response")
+    bot_msg.then(lambda: gr.MultimodalTextbox(interactive=True), None, [chat_input])
+
+    chatbot.like(print_like_dislike, None, None)
+
+demo.queue()
+demo.launch()