Update app.py

app.py

@@ -178,14 +178,20 @@ def chat_between_bots(system_message1, system_message2, max_tokens, temperature,
     
     return response1, response2, history1, history2, shared_history, f"{response1}\n\n{response2}", prosecutor_score_color, defense_score_color
 
+# Function to handle PDF uploads and display them in a gallery
+def update_pdf_gallery(pdf_files):
+    return pdf_files
+
 # Gradio interface
 with gr.Blocks(css=custom_css) as demo:
     history1 = gr.State([])
     history2 = gr.State([])
     shared_history = gr.State([])
+    pdf_files = gr.State([])
+    
     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. ")
+    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)  # Adjusted to balance response length
     temperature = gr.State(0.6)
     top_p = gr.State(0.95)
@@ -195,17 +201,21 @@ with gr.Blocks(css=custom_css) as demo:
             prosecutor_response = gr.Textbox(label="Prosecutor's Response", interactive=True)
         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)
         with gr.Column(scale=1):
             defense_score_color = gr.HTML()
     
     shared_argument = gr.Textbox(label="Case Outcome", interactive=True)
+    pdf_upload = gr.File(label="Upload Case Files (PDF)", file_types=[".pdf"], multiple=True)
+    pdf_gallery = gr.Gallery(label="PDF Gallery")
     
     submit_btn = gr.Button("Argue")
+    pdf_upload_btn = gr.Button("Update PDF Gallery")
 
     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_history, shared_argument, prosecutor_score_color, defense_score_color])
+    pdf_upload_btn.click(update_pdf_gallery, inputs=[pdf_upload], outputs=[pdf_gallery, pdf_files])
     
 if __name__ == "__main__":
     demo.launch()