Update app.py

app.py

@@ -22,14 +22,7 @@ mask_filling_pipeline = pipeline("fill-mask", model="nlpaueb/legal-bert-base-unc
 # Inference client for chat completion
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
-def respond(
-    message,
-    history: list[tuple[str, str]],
-    system_message,
-    max_tokens,
-    temperature,
-    top_p,
-):
+def respond(message, history, system_message, max_tokens, temperature, top_p):
     messages = [{"role": "system", "content": system_message}]
 
     for val in history:
@@ -51,13 +44,13 @@ def respond(
         token = message.choices[0].delta.content
         if token is not None:
             response += token
-        yield response, history + [(message, response)]
+    return response, history + [(message, response)]
 
 def generate_case_outcome(prosecutor_response, defense_response):
-    prompt = f"Give the real case details from a verified source. Detail the outcome and why."
+    prompt = f"Prosecutor's argument: {prosecutor_response}\nDefense Attorney's argument: {defense_response}\nProvide the case details from verified sources and give the outcome along with reasons."
     evaluation = ""
     for message in client.chat_completion(
-        [{"role": "system", "content": "Give the real case details from a verified source. Detail the outcome and why."},
+        [{"role": "system", "content": "Analyze the case and provide the outcome based on verified sources."},
          {"role": "user", "content": prompt}],
         max_tokens=512,
         stream=True,
@@ -82,98 +75,16 @@ def determine_winner(outcome):
     else:
         return "No clear winner"
 
-# Custom CSS for a clean layout
-custom_css = """
-body {
-    background-color: #ffffff;
-    color: #000000;
-    font-family: Arial, sans-serif;
-}
-.gradio-container {
-    max-width: 1000px;
-    margin: 0 auto;
-    padding: 20px;
-    background-color: #ffffff;
-    border: 1px solid #e0e0e0;
-    border-radius: 8px;
-    box-shadow: 0 2px 5px rgba(0, 0, 0, 0.1);
-}
-.gr-button {
-    background-color: #ffffff !important;
-    border-color: #ffffff !important;
-    color: #000000 !important;
-    margin: 5px;
-}
-.gr-button:hover {
-    background-color: #ffffff !important;
-    border-color: #004085 !important;
-}
-.gr-input, .gr-textbox, .gr-slider, .gr-markdown, .gr-chatbox {
-    border-radius: 4px;
-    border: 1px solid #ced4da;
-    background-color: #ffffff !important;
-    color: #000000 !important;
-}
-.gr-input:focus, .gr-textbox:focus, .gr-slider:focus {
-    border-color: #ffffff;
-    outline: 0;
-    box-shadow: 0 0 0 0.2rem rgba(255, 255, 255, 1.0);
-}
-#flagging-button {
-    display: none;
-}
-footer {
-    display: none;
-}
-.chatbox .chat-container .chat-message {
-    background-color: #ffffff !important;
-    color: #000000 !important;
-}
-.chatbox .chat-container .chat-message-input {
-    background-color: #ffffff !important;
-    color: #000000 !important;
-}
-.gr-markdown {
-    background-color: #ffffff !important;
-    color: #000000 !important;
-}
-.gr-markdown h1, .gr-markdown h2, .gr-markdown h3, .gr-markdown h4, .gr-markdown h5, .gr-markdown h6, .gr-markdown p, .gr-markdown ul, .gr-markdown ol, .gr-markdown li {
-    color: #000000 !important;
-}
-.score-box {
-    width: 60px;
-    height: 60px;
-    display: flex;
-    align-items: center;
-    justify-content: center;
-    font-size: 12px;
-    font-weight: bold;
-    color: black;
-    margin: 5px;
-}
-.scroll-box {
-    max-height: 200px;
-    overflow-y: scroll;
-    border: 1px solid #ced4da;
-    padding: 10px;
-    border-radius: 4px;
-}
-"""
-
 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]
-    shared_history.append(f"Prosecutor: {response1}")
-    shared_history.append(f"Defense Attorney: {response2}")
-    
-    max_length = max(len(response1), len(response2))
-    response1 = response1[:max_length]
-    response2 = response2[:max_length]
+    prosecutor_response, history1 = respond(message, history1, system_message1, max_tokens, temperature, top_p)
+    defense_response, history2 = respond(message, history2, system_message2, max_tokens, temperature, top_p)
+    shared_history.append(f"Prosecutor: {prosecutor_response}")
+    shared_history.append(f"Defense Attorney: {defense_response}")
     
-    outcome = generate_case_outcome(response1, response2)
+    outcome = generate_case_outcome(prosecutor_response, defense_response)
     winner = determine_winner(outcome)
     
-    return response1, response2, history1, history2, shared_history, outcome, winner
+    return prosecutor_response, defense_response, history1, history2, shared_history, outcome, winner
 
 def extract_text_from_pdf(pdf_file):
     text = ""
@@ -194,7 +105,6 @@ def update_pdf_gallery_and_extract_text(pdf_files):
     return pdf_files, pdf_text
 
 def get_top_10_cases():
-    # Here, I'm generating a list of 10 example cases. In a real-world scenario, you'd fetch this data from a database or another source.
     cases = [
         {"name": "Smith v. Jones", "number": "CA12345"},
         {"name": "Johnson v. State", "number": "CA67890"},
@@ -236,7 +146,7 @@ def bot(history):
         if token is not None:
             response += token
         history[-1][1] = response
-        yield history
+    return history
 
 def print_like_dislike(x: gr.LikeData):
     print(x.index, x.value, x.liked)
@@ -251,6 +161,84 @@ def ask_about_case_outcome(shared_history, question):
     result = qa_pipeline(question=question, context=shared_history)
     return result['answer']
 
+# Custom CSS for a clean layout
+custom_css = """
+body {
+    background-color: #ffffff;
+    color: #000000;
+    font-family: Arial, sans-serif;
+}
+.gradio-container {
+    max-width: 1000px;
+    margin: 0 auto;
+    padding: 20px;
+    background-color: #ffffff;
+    border: 1px solid #e0e0e0;
+    border-radius: 8px;
+    box-shadow: 0 2px 5px rgba(0, 0, 0, 0.1);
+}
+.gr-button {
+    background-color: #ffffff !important;
+    border-color: #ffffff !important;
+    color: #000000 !important;
+    margin: 5px;
+}
+.gr-button:hover {
+    background-color: #ffffff !important;
+    border-color: #004085 !important;
+}
+.gr-input, .gr-textbox, .gr-slider, .gr-markdown, .gr-chatbox {
+    border-radius: 4px;
+    border: 1px solid #ced4da;
+    background-color: #ffffff !important;
+    color: #000000 !important;
+}
+.gr-input:focus, .gr-textbox:focus, .gr-slider:focus {
+    border-color: #ffffff;
+    outline: 0;
+    box-shadow: 0 0 0 0.2rem rgba(255, 255, 255, 1.0);
+}
+#flagging-button {
+    display: none;
+}
+footer {
+    display: none;
+}
+.chatbox .chat-container .chat-message {
+    background-color: #ffffff !important;
+    color: #000000 !important;
+}
+.chatbox .chat-container .chat-message-input {
+    background-color: #ffffff !important;
+    color: #000000 !important;
+}
+.gr-markdown {
+    background-color: #ffffff !important;
+    color: #000000 !important;
+}
+.gr-markdown h1, .gr-markdown h2, .gr-markdown h3, .gr-markdown h4, .gr-markdown h5, .gr-markdown h6, .gr-markdown p, .gr-markdown ul, .gr-markdown ol, .gr-markdown li {
+    color: #000000 !important;
+}
+.score-box {
+    width: 60px;
+    height: 60px;
+    display: flex;
+    align-items: center;
+    justify-content: center;
+    font-size: 12px;
+    font-weight: bold;
+    color: black;
+    margin: 5px;
+}
+.scroll-box {
+    max-height: 200px;
+    overflow-y: scroll;
+    border: 1px solid #ced4da;
+    padding: 10px;
+    border-radius: 4px;
+}
+"""
+
 with gr.Blocks(css=custom_css) as demo:
     history1 = gr.State([])
     history2 = gr.State([])