Update app.py

app.py

@@ -50,7 +50,7 @@ def respond(
         yield response, history + [(message, response)]
 
 def generate_case_outcome(prosecutor_response, defense_response):
-    prompt = "Provide details on who won the case and why. Provide reasons for your decision and provide a link to the source of the case."
+    prompt = f"Prosecutor's arguments: {prosecutor_response}\n\nDefense's arguments: {defense_response}\n\nProvide details on who won the case and why. Provide reasons for your decision and provide a link to the source of the case."
     evaluation = ""
     for message in client.chat_completion(
         [{"role": "system", "content": "You are a legal expert evaluating the details of the case presented by the prosecution and the defense."},
@@ -66,14 +66,11 @@ def generate_case_outcome(prosecutor_response, defense_response):
     return evaluation
 
 def determine_winner(outcome):
-    if "Prosecutor" in outcome and "Defense" in outcome:
-        if "Prosecutor" in outcome.split().count("Prosecutor") > outcome.split().count("Defense"):
-            return "Prosecutor Wins"
-        else:
-            return "Defense Wins"
-    elif "Prosecutor" in outcome:
+    prosecutor_count = outcome.split().count("Prosecutor")
+    defense_count = outcome.split().count("Defense")
+    if prosecutor_count > defense_count:
         return "Prosecutor Wins"
-    elif "Defense" in outcome:
+    elif defense_count > prosecutor_count:
         return "Defense Wins"
     else:
         return "No clear winner"
@@ -202,7 +199,7 @@ def update_pdf_gallery_and_extract_text(pdf_files):
     return pdf_files, pdf_text
 
 def get_top_10_cases():
-    prompt = " 10 high-profile legal cases that have received significant media attention and are currently ongoing. Just a list of case names and numbers"
+    prompt = "List 10 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."},