Update app.py

app.py

@@ -185,8 +185,8 @@ def chat_between_bots(system_message1, system_message2, max_tokens, temperature,
     
     return response1, response2, history1, history2, shared_history, outcome
 
-def get_top_10_cases():
-    prompt = "List 10 high-profile legal cases that have received significant media attention and are currently ongoing. Just a list of case names and numbers."
+def get_top_5_cases():
+    prompt = "List 5 random high-profile legal cases that have received significant media attention and are currently ongoing. Just a list of case names and numbers."
     response = ""
     for message in client.chat_completion(
         [{"role": "system", "content": "You are a legal research expert, able to provide information about high-profile legal cases."},
@@ -221,11 +221,11 @@ def ask_about_case_outcome(shared_history, question):
     prompt = f"Case Outcome: {shared_history}\n\nQuestion: {question}\n\nAnswer:"
     response = ""
     for message in client.chat_completion(
-        [{"role": "system", "content": "You are a legal expert answering questions based on the case outcome provided."},
+        [{"role": "system", "content": "Give the real case details of the case being argued, from a verifed source."},
          {"role": "user", "content": prompt}],
         max_tokens=512,
         stream=True,
-        temperature=0.6,
+        temperature=0.7,
         top_p=0.95,
     ):
         token = message.choices[0].delta.content
@@ -242,15 +242,15 @@ with gr.Blocks(css=custom_css) as demo:
     with gr.Tab("Argument Evaluation"):
         with gr.Row():
             with gr.Column(scale=1):
-                top_10_btn = gr.Button("Give me the top 10 cases")
-                top_10_output = gr.Textbox(label="Top 10 Cases", interactive=False, elem_classes=["scroll-box"])
-                top_10_btn.click(get_top_10_cases, outputs=top_10_output)
+                top_5_btn = gr.Button("Give me the top 5 cases")
+                top_5_output = gr.Textbox(label="Top 5 Cases", interactive=False, elem_classes=["scroll-box"])
+                top_5_btn.click(get_top_5_cases, outputs=top_5_output)
             with gr.Column(scale=2):
                 message = gr.Textbox(label="Case to Argue")
                 system_message1 = gr.State("You are an expert Prosecutor. Give your best arguments for the case on behalf of the prosecution.")
                 system_message2 = gr.State("You are an expert Defense Attorney. Give your best arguments for the case on behalf of the Defense.")
                 max_tokens = gr.State(512)
-                temperature = gr.State(0.6)
+                temperature = gr.State(0.5)
                 top_p = gr.State(0.95)
                 
                 with gr.Row():
@@ -264,7 +264,7 @@ with gr.Blocks(css=custom_css) as demo:
                     with gr.Column(scale=1):
                         defense_score_color = gr.HTML()
                 
-                outcome = gr.Textbox(label="Outcome", interactive=False, elem_classes=["scroll-box"])
+                outcome = gr.Textbox(label="Outcome", interactive=True, elem_classes=["scroll-box"])
                 
                 with gr.Row():
                     submit_btn = gr.Button("Argue")