nbuechner/matrix-nsfwbot
nbuechner/matrix-nsfwbot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add sender to report message

Browse source
This commit is contained in:
Nils Büchner 2024-09-27 15:16:39 +02:00
parent 45a085075c
commit 49a0b9dd89
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
nsfwbot.py
View file

@ -116,7 +116,7 @@ class NSFWModelPlugin(Plugin):
return return
results = await self.process_images([evt.content.url]) results = await self.process_images([evt.content.url])
matrix_to_url = self.create_matrix_to_url(evt.room_id, evt.event_id) matrix_to_url = self.create_matrix_to_url(evt.room_id, evt.event_id)
response = self.format_response(results, matrix_to_url) response = self.format_response(results, matrix_to_url, evt.sender)
await self.send_responses(evt, response, results) await self.send_responses(evt, response, results)
except Exception as e: except Exception as e:
self.log.error(f"Error handling image message: {e}") self.log.error(f"Error handling image message: {e}")
@ -156,7 +156,7 @@ class NSFWModelPlugin(Plugin):
for _, temp_filename in temp_files: for _, temp_filename in temp_files:
os.remove(temp_filename) os.remove(temp_filename)
def format_response(self, results: dict, matrix_to_url: str) -> str: def format_response(self, results: dict, matrix_to_url: str, sender: str) -> str:
""" """
Format the response message based on the results. Format the response message based on the results.
@ -171,7 +171,8 @@ class NSFWModelPlugin(Plugin):
detection_info = ( detection_info = (
f"{mxc_url} contains {detection['class']} " f"{mxc_url} contains {detection['class']} "
f"with a score of {detection['score']:.2f} " f"with a score of {detection['score']:.2f} "
f"in {matrix_to_url}" f"in {matrix_to_url} "
f"by {sender}"
) )
response_parts.append(detection_info) response_parts.append(detection_info)
break break