Update app.py

app.py

@@ -9,27 +9,25 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 from transformers import pipeline
 from datasets import load_dataset
-import time
 import fitz  # PyMuPDF
 
-dataset = load_dataset("ibunescu/qa_legal_dataset_train")
+client = InferenceClient()
 
-    return evaluation
+dataset = load_dataset("ibunescu/qa_legal_dataset_train")
 
 def score_argument_from_outcome(outcome, argument):
-
+    prosecutor_score = 0
     if "Prosecutor" in outcome:
         prosecutor_score = outcome.count("Prosecutor") * 2
         if "won" in outcome and "Prosecutor" in outcome:
-}
-"""
-
+            prosecutor_score += 10
+    return prosecutor_score
 
 def chat_between_bots(system_message1, system_message2, max_tokens, temperature, top_p, history1, history2, shared_history, message):
     response1, history1 = list(respond(message, history1, system_message1, max_tokens, temperature, top_p))[-1]
     response2, history2 = list(respond(message, history2, system_message2, max_tokens, temperature, top_p))[-1]
     
-    return response1, response2, history1, history2, shared_history, outcome, prosecutor_score_color, defense_score_color
+    return response1, response2, history1, history2, shared_history
 
 def extract_text_from_pdf(pdf_file):
     text = ""
@@ -62,8 +60,8 @@ def update_pdf_gallery_and_extract_text(pdf_files):
     return pdf_files, pdf_text
 
 def add_message(history, message):
-    for x in message["files"]:
-    return history, gr.MultimodalTextbox(value=None, interactive=False)
+    history.append(message)
+    return history, gr.Textbox(value=None, interactive=False)
 
 def bot(history):
     system_message = "You are a helpful assistant."
@@ -96,6 +94,13 @@ def reset_conversation():
 def save_conversation(history1, history2, shared_history):
     return history1, history2, shared_history
 
+custom_css = """
+.scroll-box {
+    max-height: 400px;
+    overflow-y: auto;
+}
+"""
+
 with gr.Blocks(css=custom_css) as demo:
     history1 = gr.State([])
     history2 = gr.State([])
@@ -105,8 +110,18 @@ with gr.Blocks(css=custom_css) as demo:
     
     with gr.Tab("Argument Evaluation"):
         message = gr.Textbox(label="Case to Argue")
+        system_message1 = "System message for bot 1"
+        system_message2 = "System message for bot 2"
+        max_tokens = 150
+        temperature = 0.6
+        top_p = 0.95
         
+        prosecutor_response = gr.Textbox(label="Prosecutor Response", interactive=False)
+        defense_response = gr.Textbox(label="Defense Response", interactive=False)
+        prosecutor_score_color = gr.Textbox(label="Prosecutor Score Color", interactive=False)
+        defense_score_color = gr.Textbox(label="Defense Score Color", interactive=False)
         shared_argument = gr.Textbox(label="Case Outcome", interactive=True)
+        
         submit_btn = gr.Button("Argue")
         clear_btn = gr.Button("Clear and Reset")
         save_btn = gr.Button("Save Conversation")
@@ -123,10 +138,12 @@ with gr.Blocks(css=custom_css) as demo:
         pdf_answer = gr.Textbox(label="Answer", interactive=False, elem_classes=["scroll-box"])
         pdf_upload_btn = gr.Button("Update PDF Gallery")
         pdf_ask_btn = gr.Button("Ask")
-
+        
         pdf_upload_btn.click(update_pdf_gallery_and_extract_text, inputs=[pdf_upload], outputs=[pdf_gallery, pdf_text])
         pdf_text.change(fn=lambda x: x, inputs=pdf_text, outputs=pdf_view)
         pdf_ask_btn.click(ask_about_pdf, inputs=[pdf_text, pdf_question], outputs=pdf_answer)
     
     with gr.Tab("Chatbot"):
-        chatbot = gr.Chatbot(
+        chatbot = gr.Chatbot()
+        
+demo.launch()