File size: 1,323 Bytes
c3d8cb2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
import requests
def rewrite_feedback(feedback: str, tone: str, api_key: str) -> str:
system_prompt = (
"You are an assistant that rewrites workplace feedback messages to improve tone and clarity. "
"Depending on the selected tone, adjust the response accordingly:\n"
"- Empathetic: Use gentle, understanding language.\n"
"- Constructive: Focus on solutions and next steps.\n"
"- Managerial: Be direct but professional and goal-oriented."
)
user_prompt = f"Rewrite the following feedback in a {tone} tone:\n\n\"{feedback}\""
headers = {
"Authorization": f"Bearer {api_key}",
"HTTP-Referer": "https://feedback-rewriter.streamlit.app", # replace with your actual Streamlit app URL later
"X-Title": "Feedback Rewriter Assistant"
}
data = {
"model": "mistralai/mistral-7b-instruct",
"messages": [
{"role": "system", "content": system_prompt},
{"role": "user", "content": user_prompt}
]
}
response = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers, json=data)
if response.status_code == 200:
return response.json()["choices"][0]["message"]["content"].strip()
else:
return f"Error: {response.status_code} - {response.text}"
|