Update app.py

app.py

@@ -7,7 +7,7 @@ client = InferenceClient(
     "mistralai/Mixtral-8x7B-Instruct-v0.1"
 )
 
-system_prompt = """You are Mixtral, a professional and polite use of force writer. You will receive patient name, patient number, time, date, hospital ward, officer name and where they saw patient when officers arrived and any aditional information,add the additional information in between and always write polit and calm and professional report. you always use this sample as guide to write the report : 
+system_prompt = """You are Mixtral, a professional and polite use of force writer. You will receive patient name, patient number, time, date, hospital ward, officer name, PI Position and where they saw patient when officers arrived and any aditional information,add the additional information in between and always write polit and calm and professional report. you always use this sample as guide to write the report : 
 LOCATION: G3 ward, Addenbrookes Hospital
 
 DATE:[date]
@@ -28,13 +28,13 @@ def format_prompt(message, history):
   return prompt
 
 def generate(
-    time,ward,p,pn,on,loc,adinfo, history='', temperature=0.9, max_new_tokens=1000, top_p=0.95, repetition_penalty=1.0,
+    time,ward,p,pn,on,loc,pipos,adinfo, history='', temperature=0.9, max_new_tokens=1000, top_p=0.95, repetition_penalty=1.0,
 ):
     # Month abbreviation, day and year	
     d4 = today.strftime("%b-%d-%Y")
     print("d4 =", d4)
 
-    prompt="Date is:"+d4+" time was:"+time+ " Ward was: "+ward+", Addenbrookes Hospital "+" Patient Name is:"+p+" Patient Number is: "+pn+" Officer Name is: "+on +" Patient Location was: "+loc +" Additional information to better understand the incident: "+ adinfo
+    prompt="Date is:"+d4+" time was:"+time+ " Ward was: "+ward+", Addenbrookes Hospital "+" Patient Name is:"+p+" Patient Number is: "+pn+" Officer Name is: "+on +" Patient Location was: "+loc +" PI Postion was: "+pipos+" Additional information to better understand the incident: "+ adinfo
     temperature = float(temperature)
     if temperature < 1e-2:
         temperature = 1e-2
@@ -103,7 +103,7 @@ additional_inputs=[
     )
 ]
 
-examples=[["10:30", "D9", "Allen Mark", "1581760", "Bilal Sardar", "outside ward","had spit on me" ],
+examples=[["10:30", "D9", "Allen Mark", "1581760", "Bilal Sardar", "outside ward","Right Arm,"had spit on me" ],
          ]
 
 css="""
@@ -120,7 +120,7 @@ gr.Interface(
     fn=generate,
     #chatbot=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, bubble_full_width=False, avatar_images=[None, "logo.jpg"]),
     #additional_inputs=additional_inputs,
-    inputs=[gr.Textbox(label='Time e.g 15:30'),gr.Textbox(label='Hospital ward'),gr.Textbox(label='Patient Name'),gr.Textbox(label='Patient Number (seven digit number)'),gr.Textbox(label='Officer Name'),gr.Textbox(label='Patient Location'),gr.Textbox(label='Any Additional Information? (optional)')],
+    inputs=[gr.Textbox(label='Time e.g 15:30'),gr.Textbox(label='Hospital ward'),gr.Textbox(label='Patient Name'),gr.Textbox(label='Patient Number (seven digit number)'),gr.Textbox(label='Officer Name'),gr.Textbox(label='Patient Location'),gr.inputs.Radio(["Left Arm", "Right Arm","Legs"], type="value", label='PI Position'),gr.Textbox(label='Any Additional Information? (optional)')],
     examples=examples,
     outputs=gr.Textbox(label='Use Of Force Report'),
     title="Use Of Force",