Update app.py

app.py

@@ -174,7 +174,25 @@ def chat_between_bots(system_message1, system_message2, max_tokens, temperature,
     prosecutor_score_color = f"<div class='score-box' style='background-color:{prosecutor_color};'>Score: {score1}</div>"
     defense_score_color = f"<div class='score-box' style='background-color:{defense_color};'>Score: {score2}</div>"
     
-    return response1, response2, history1, history2, shared_history, f"{response1}\n\n{response2}", prosecutor_score_color, defense_score_color
+    outcome = generate_case_outcome(response1, response2)
+    
+    return response1, response2, history1, history2, shared_history, outcome, prosecutor_score_color, defense_score_color
+
+def generate_case_outcome(prosecutor_response, defense_response):
+    prompt = f"Prosecutor's Argument: {prosecutor_response}\nDefense Attorney's Argument: {defense_response}\n\nEvaluate both arguments and determine who won the case. Provide reasons for your decision."
+    evaluation = ""
+    for message in client.chat_completion(
+        [{"role": "system", "content": "You are a legal expert evaluating the arguments presented by the prosecution and the defense."},
+         {"role": "user", "content": prompt}],
+        max_tokens=512,
+        stream=True,
+        temperature=0.6,
+        top_p=0.95,
+    ):
+        token = message.choices[0].delta.content
+        if token is not None:
+            evaluation += token
+    return evaluation
 
 def update_pdf_gallery(pdf_files):
     return pdf_files
@@ -213,14 +231,12 @@ with gr.Blocks(css=custom_css) as demo:
         
         with gr.Row():
             with gr.Column(scale=4):
-                prosecutor_response = gr.Textbox(label="Prosecutor's Response", interactive=True)
-                prosecutor_response= gr.style(container=True).add_class("scroll-box")
+                prosecutor_response = gr.Textbox(label="Prosecutor's Response", interactive=True, elem_classes=["scroll-box"])
             with gr.Column(scale=1):
                 prosecutor_score_color = gr.HTML()
             
             with gr.Column(scale=4):
-                defense_response = gr.Textbox(label="Defense Attorney's Response", interactive=True)
-                defense_response.style(container=True).add_class("scroll-box")
+                defense_response = gr.Textbox(label="Defense Attorney's Response", interactive=True, elem_classes=["scroll-box"])
             with gr.Column(scale=1):
                 defense_score_color = gr.HTML()