Update app.py

app.py

@@ -87,9 +87,10 @@ def generate(
     #prompt_finale = prompt_template(materiali, message)
     conversation = []
     conversation.append({"role": "system", "content": 
-                         '''Sei un an assistente AI di nome 'AvvoChat' specializzato nel rispondere a domande riguardanti la legge Italiana. 
+                         '''Sei un an assistente AI di nome 'AvvoChat' specializzato nel rispondere la legge Italiana. 
+                         Se le domande non riguardano questioni legali astieniti dal rispondere e scrivi "Sono specializzato in domande di tipo legale: non sono accurato su questo tipo di domande".
                          Rispondi in lingua italiana in modo chiaro, semplice ed esaustivo alle domande che ti vengono fornite.
-                         Le risposte devono essere sintetiche e chiare di massimo 100 parole o anche più corte. 
+                         Le risposte devono essere chiare e semplici con argomentazioni valide e puntuali. 
                          Firmati alla fine di ogni risposta '-AvvoChat'.'''})
     for user, assistant in chat_history:
         conversation.extend([{"role": "user", "content": user}, {"role": "assistant", "content": assistant}])
@@ -98,7 +99,7 @@ def generate(
     input_ids = tokenizer.apply_chat_template(conversation, return_tensors="pt")
     if input_ids.shape[1] > MAX_INPUT_TOKEN_LENGTH:
         input_ids = input_ids[:, -MAX_INPUT_TOKEN_LENGTH:]
-        gr.Warning(f"Trimmed input from conversation as it was longer than {MAX_INPUT_TOKEN_LENGTH} tokens.")
+        gr.Warning(f"Conversazione troppo lunga: sforati i {MAX_INPUT_TOKEN_LENGTH} tokens.")
     input_ids = input_ids.to(model.device)
 
     streamer = TextIteratorStreamer(tokenizer, timeout=30.0, skip_prompt=True, skip_special_tokens=True)