Update app.py

app.py

@@ -221,6 +221,22 @@ def update_pdf_gallery_and_extract_text(pdf_files):
         pdf_text = ""
     return pdf_files, pdf_text
 
+def get_top_10_cases():
+    prompt = "Give me a list of the current top 10 cases in the country being discussed by the top lawyers in the country."
+    response = ""
+    for message in client.chat_completion(
+        [{"role": "system", "content": "You are a legal expert providing information about top legal cases."},
+         {"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:
+            response += token
+    return response
+
 def add_message(history, message):
     for x in message["files"]:
         history.append(((x,), None))
@@ -265,34 +281,41 @@ with gr.Blocks(css=custom_css) as demo:
     shared_history = gr.State([])
     pdf_files = gr.State([])
     pdf_text = gr.State("")
-    
+    top_10_cases = gr.State("")
+
     with gr.Tab("Argument Evaluation"):
-        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)
-        top_p = gr.State(0.95)
-        
         with gr.Row():
-            with gr.Column(scale=4):
-                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, elem_classes=["scroll-box"])
             with gr.Column(scale=1):
-                defense_score_color = gr.HTML()
-        
-        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")
-        
-        submit_btn.click(chat_between_bots, inputs=[system_message1, system_message2, max_tokens, temperature, top_p, history1, history2, shared_history, message], outputs=[prosecutor_response, defense_response, history1, history2, shared_argument, prosecutor_score_color, defense_score_color])
-        clear_btn.click(reset_conversation, outputs=[history1, history2, shared_history, prosecutor_response, defense_response, shared_argument])
-        save_btn.click(save_conversation, inputs=[history1, history2, shared_history], outputs=[history1, history2, shared_history])
+                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)
+            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)
+                top_p = gr.State(0.95)
+                
+                with gr.Row():
+                    with gr.Column(scale=4):
+                        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, elem_classes=["scroll-box"])
+                    with gr.Column(scale=1):
+                        defense_score_color = gr.HTML()
+                
+                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")
+                
+                submit_btn.click(chat_between_bots, inputs=[system_message1, system_message2, max_tokens, temperature, top_p, history1, history2, shared_history, message], outputs=[prosecutor_response, defense_response, history1, history2, shared_argument, prosecutor_score_color, defense_score_color])
+                clear_btn.click(reset_conversation, outputs=[history1, history2, shared_history, prosecutor_response, defense_response, shared_argument])
+                save_btn.click(save_conversation, inputs=[history1, history2, shared_history], outputs=[history1, history2, shared_history])
     
     with gr.Tab("PDF Management"):
         pdf_upload = gr.File(label="Upload Case Files (PDF)", file_types=[".pdf"])