Update app.py

app.py

@@ -120,18 +120,23 @@ with gr.Blocks(css=custom_css) as demo:
     with gr.Column():
         gr.Markdown("# Court Argument Simulation\n### Provide Initial Case Details")
         case_details = gr.Textbox(lines=5, placeholder="Enter initial case details here...")
+        evidence = gr.Textbox(lines=3, placeholder="Enter evidence details here...")
+        witness_statements = gr.Textbox(lines=3, placeholder="Enter witness statements here...")
+        legal_references = gr.Textbox(lines=3, placeholder="Enter legal references here...")
+        
         prosecution_argument = gr.Textbox(lines=10, placeholder="Prosecution's Argument...")
         defense_argument = gr.Textbox(lines=10, placeholder="Defense's Argument...")
         
-        def run_simulation(case_details):
-            prosecution_arg = generate_prosecution_argument(case_details)
+        def run_simulation(case_details, evidence, witness_statements, legal_references):
+            full_case_details = f"Case Details: {case_details}\n\nEvidence: {evidence}\n\nWitness Statements: {witness_statements}\n\nLegal References: {legal_references}"
+            prosecution_arg = generate_prosecution_argument(full_case_details)
             defense_arg = generate_defense_argument(prosecution_arg)
             return prosecution_arg, defense_arg
         
         simulate_btn = gr.Button("Start Argument Simulation")
-        simulate_btn.click(run_simulation, inputs=[case_details], outputs=[prosecution_argument, defense_argument])
+        simulate_btn.click(run_simulation, inputs=[case_details, evidence, witness_statements, legal_references], outputs=[prosecution_argument, defense_argument])
         
         clear_btn = gr.Button("Clear")
-        clear_btn.click(lambda: ("", "", ""), None, [case_details, prosecution_argument, defense_argument])
+        clear_btn.click(lambda: ("", "", "", "", "", ""), None, [case_details, evidence, witness_statements, legal_references, prosecution_argument, defense_argument])
 
 demo.launch()