fix generate_view_url call

charliettaylor · Sep 1, 2024 · ef3ebf1 · ef3ebf1
1 parent dfa882f
commit ef3ebf1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/Client.py b/Client.py
@@ -102,6 +102,10 @@ def handle_image(self, from_: str, url: str) -> None:
             return
 
         prompt = self.db.get_current_prompt()
+
+        if prompt is None:
+            return
+
         exists = self.db.get_submission_status(user.hash, prompt.id)
         if exists:
             self.send_message(from_, ALREADY_SUBMITTED)
@@ -112,7 +116,7 @@ def handle_image(self, from_: str, url: str) -> None:
             self.send_message(from_, FAILED_PIC_SAVE)
         else:
             self.send_message(
-                from_, VIEW_SUBMISSIONS.format(self.generate_view_url(user.hash))
+                from_, VIEW_SUBMISSIONS.format(self.generate_view_url(user.hash, prompt.id))
             )
 
     def handle_admin_message(self, text: str) -> None: