Update app.py

app.py

@@ -259,6 +259,9 @@ def simple_chat(message: dict, temperature: float = 0.8, max_length: int = 4096,
 
         gen_kwargs = {**input_ids, **generate_kwargs}
 
+        for entry in conversation:
+            print(f"Role: {entry['role']}, Content: {entry.get('content', '')}")
+
         with torch.no_grad():
             generated_ids = model.generate(input_ids['input_ids'], **generate_kwargs)
             generated_text = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
@@ -273,7 +276,7 @@ def simple_chat(message: dict, temperature: float = 0.8, max_length: int = 4096,
         print(generated_text_cleaned)
 
 
-        return PlainTextResponse(generated_text)
+        return PlainTextResponse(generated_text_cleaned)
     except Exception as e:
         return PlainTextResponse(f"Error: {str(e)}")