Update poept.py

Saikyo0 · Jul 21, 2024 · db81474 · db81474
1 parent d15b921
commit db81474
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/poept/poept.py b/poept/poept.py
@@ -217,10 +217,10 @@ def ask(self,
                 msg = self.driver.find_element(By.XPATH, f"(//div[@class='{self.msg_element[1:]}'])[last()]")
                 if msg.get_attribute("data-complete") == "true": break
 
+            self.response = '\n'.join(msg.text.split('\n')[2:])
             if img_output:
-                self.response = msg.find_element(By.CSS_SELECTOR, self.msg_image).get_attribute("src")
+                self.response += msg.find_element(By.CSS_SELECTOR, self.msg_image).get_attribute("src")
 
-            self.response += '\n'.join(msg.text.split('\n')[2:])
             self.status = "ready"
             return self.response