AppState product_name

app.py

@@ -15,6 +15,9 @@ app = FastAPI()
 async def read_root():
     return {"message": "Hello, World!"}
 
+class AppState:
+    product_name = None
+
 
 class LoginException(HTTPException):
     """Custom exception for login failures."""
@@ -55,8 +58,8 @@ async def siloshot_making(page, fabric_name):
     # Zbieranie nazwy produktu z nagłówka h1 o klasie ng-star-inserted
     product_name_element = soup.find('h1', class_='ng-star-inserted')
     if product_name_element:
-        product_name = product_name_element.text.strip()
-        print(f'Product name collected: {product_name}')
+        AppState.product_name = product_name_element.text.strip()
+        print(f'Product name collected: {AppState.product_name}')
     else:
         raise Exception('Product name not found in the page.')
 
@@ -153,7 +156,7 @@ async def download_images(img_urls, fabric_name):
                 if angle_match:
                     # Create a file name using the product name and the angle
                     angle_value = angle_match.group(1)
-                    img_name = f"{angle_value}_{fabric_name}.png"
+                    img_name = f"{AppState.product_name}_{angle_value}_{fabric_name}.png"
                 else:
                     # Default file name if the angle is not found
                     img_name = "product_unknown.jpg"
@@ -200,7 +203,7 @@ async def process_images(username, password, product_url, fabric_name):
 
         img = await siloshot_making(page, fabric_name)
         img_urls = generate_urls(img, start_angle=0, end_angle=360, step=10)
-        zip_buffer = await download_images(img_urls, fabric_name)
+        zip_buffer = await download_images(img_urls, fabric_name, AppState.product_name)
         await browser.close()
         return zip_buffer