fix paths and add cache for images

.gitignore

@@ -129,3 +129,5 @@ dmypy.json
 
 # Pyre type checker
 .pyre/
+/backend/services/prompt_extractor.py.cache.db
+/backend/services/stable_diffusion_consumer.py.cache.db

backend/main.py

@@ -4,7 +4,7 @@
 from os import path
 from typing import Dict
 
-from backend.services.prompt_extractor import PromptExtractor
+from services.prompt_extractor import PromptExtractor
 from fetch_images import fetch_image
 
 test_str = "There was a miller whose only inheritance to his three sons was his mill, his donkey, and his cat. The division was soon made. They hired neither a clerk nor an attorney, for they would have eaten up all the poor patrimony. The eldest took the mill, the second the donkey, and the youngest nothing but the cat. + oil painting"

backend/serve.py

@@ -28,9 +28,9 @@ async def root():
 
 
 @app.post("/imageFromPromt")
-async def image_from_promt(payload: PromptPayload) -> StreamingResponse:
+def image_from_promt(payload: PromptPayload) -> StreamingResponse:
     consumer = StableDiffusionConsumer()
-    image_bytes = next(consumer.fetch_image(payload.prompt))
+    image_bytes = consumer.fetch_image(payload.prompt)
 
     return StreamingResponse(consumer.parse_to_bytesio(image_bytes), media_type="image/png")
 

backend/test/prompt_extractor_test.py

@@ -1,7 +1,7 @@
 import unittest
 from os import path
 
-from backend.services.prompt_extractor import PromptExtractor
+from services.prompt_extractor import PromptExtractor
 
 
 class MyTestCase(unittest.TestCase):