Add timeout message

Add timeout message

Author: JE-Chen

re_edge_gpt/image_genearation.py

@@ -169,7 +169,7 @@ def get_images(self, prompt: str, timeout: int = 200, max_generate_time_sec: int
                 time.sleep(1)
                 time_sec = time_sec + 1
                 if time_sec >= max_generate_time_sec:
-                    raise TimeoutError
+                    raise TimeoutError("Out of generate time")
                 continue
             else:
                 break
@@ -344,7 +344,7 @@ async def get_images(self, prompt: str, timeout: int = 200, max_generate_time_se
             await asyncio.sleep(1)
             time_sec = time_sec + 1
             if time_sec >= max_generate_time_sec:
-                raise TimeoutError
+                raise TimeoutError("Out of generate time")
             continue
         # Use regex to search for src=""
         image_links = regex.findall(r'src="([^"]+)"', content)

test/unit_test/test_generate_image/test_generation_manual_async_loop.py

@@ -15,8 +15,8 @@
 
 # Generate image list async
 async def test_generate_image_async():
-    print("Generate 🐱")
-    image_list = await async_gen.get_images("🐱")
+    print("Generate Pigeon")
+    image_list = await async_gen.get_images("Pigeon")
     print(image_list)
 
 
@@ -31,6 +31,6 @@ async def test_generate_image_async():
             # Remove dir
             shutil.rmtree(test_output_dir)
     except Exception as error:
-        raise error
+        print(repr(error), file=sys.stderr)
     finally:
         sys.exit(0)