Update app.py

app.py

@@ -53,12 +53,16 @@ def generate_text(input_text, max_new_tokens=512, temperature=1, system_prompt="
         pad_token_id=tokenizer.eos_token_id
     )
     generated_text = output[0]['generated_text']
+    generated_text_color = 'blue'
 
     split_tentative = generated_text.split("### Risposta:")
     if len(split_tentative) > 1:
         generated_text = split_tentative[1]
+    elif '### Istruzione:' in split_tentative[0]:
+        generated_text = "Spiacente, non sono in grado di rispondere."
+        generated_text_color = 'red'
     
-    return f"<span>{input_text}</span><b style='color: blue;'>{generated_text}</b>"
+    return f"<span>{input_text}</span><b style='color: {generated_text_color};'>{generated_text}</b>"
 
 # Create the Gradio interface
 input_text = gr.Textbox(lines=2, placeholder="Enter your request here...", label="Input Text")