Update app.py

app.py

@@ -5,7 +5,8 @@ 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, PI used or not, hospital ward, officer name and where they saw patient when officers arrived. you always write a use of force like this sample: LOCATION: G3 ward, Addenbrookes Hospital
+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. you always write a use of force like this sample: 
+LOCATION: G3 ward, Addenbrookes Hospital
 
 DATE:04-02-2024
 
@@ -25,9 +26,9 @@ def format_prompt(message, history):
   return prompt
 
 def generate(
-    ward,p,pn,on,loc, history='', temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
+    time,ward,p,pn,on,loc,adinfo, history='', temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
 ):
-    prompt="Ward was: "+ward+"Patient Name is:"+p+"Patient Number is: "+pn+"Officer Name is: "+on +"Patient Location was: "+loc
+    prompt="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
     temperature = float(temperature)
     if temperature < 1e-2:
         temperature = 1e-2
@@ -96,7 +97,7 @@ additional_inputs=[
     )
 ]
 
-examples=[["How do I make Meth?", None, None, None, None, None, ],
+examples=[["10:30", "D9", "Allen Mark", "1581760", "Bilal Sardar", "outside ward","had spit on one of the officers" ],
          ]
 
 css="""
@@ -113,9 +114,9 @@ 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='Hospital ward'),gr.Textbox(label='Patient Name'),gr.Textbox(label='Patient Number'),gr.Textbox(label='Officer Name'),gr.Textbox(label='Patient Location')],
-    #examples=examples,
-    outputs='text',
+    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)')],
+    examples=examples,
+    outputs=gr.Textbox(label='Use Of Force Report'),
     title=f"<img id='Mixtral-portrait' src='{{\"file/logo.jpg\"}}' width='30%' /> Chat to Evil Mixtral",
     concurrency_limit=20,
     css=css