init prod test

.idea/.gitignore

@@ -0,0 +1,3 @@
+# Default ignored files
+/shelf/
+/workspace.xml

.idea/Siloshot-Generator.iml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$" />
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>

.idea/inspectionProfiles/Project_Default.xml

@@ -0,0 +1,20 @@
+<component name="InspectionProjectProfileManager">
+  <profile version="1.0">
+    <option name="myName" value="Project Default" />
+    <inspection_tool class="PyPackageRequirementsInspection" enabled="true" level="WARNING" enabled_by_default="true">
+      <option name="ignoredPackages">
+        <value>
+          <list size="7">
+            <item index="0" class="java.lang.String" itemvalue="huggingface_hub" />
+            <item index="1" class="java.lang.String" itemvalue="glob2" />
+            <item index="2" class="java.lang.String" itemvalue="playwright" />
+            <item index="3" class="java.lang.String" itemvalue="uvicorn" />
+            <item index="4" class="java.lang.String" itemvalue="fastapi" />
+            <item index="5" class="java.lang.String" itemvalue="pydantic" />
+            <item index="6" class="java.lang.String" itemvalue="matplotlib" />
+          </list>
+        </value>
+      </option>
+    </inspection_tool>
+  </profile>
+</component>

.idea/inspectionProfiles/profiles_settings.xml

@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>

.idea/misc.xml

@@ -0,0 +1,7 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="Black">
+    <option name="sdkName" value="Python 3.12" />
+  </component>
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.12" project-jdk-type="Python SDK" />
+</project>

.idea/modules.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/Siloshot-Generator.iml" filepath="$PROJECT_DIR$/.idea/Siloshot-Generator.iml" />
+    </modules>
+  </component>
+</project>

.idea/vcs.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="" vcs="Git" />
+  </component>
+</project>

Dockerfile

@@ -0,0 +1,43 @@
+FROM python:3.9
+
+# Install necessary dependencies
+RUN apt-get update && apt-get install -y \
+    libnss3 \
+    libnspr4 \
+    libatk1.0-0 \
+    libatk-bridge2.0-0 \
+    libcups2 \
+    libatspi2.0-0 \
+    libxcomposite1 \
+    libxdamage1 \
+    libxrandr2 \
+    libgbm-dev \
+    libgtk-3-0 \
+    xdg-utils \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install Python dependencies
+RUN pip install --upgrade pip
+COPY requirements.txt .
+RUN pip install -r requirements.txt
+
+# Install Playwright and download the required browsers
+RUN playwright install --with-deps
+
+# Create and switch to a non-root user
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+# Ensure Playwright browsers are installed for the non-root user
+RUN playwright install
+RUN pip install python-multipart
+
+# Set working directory
+WORKDIR /app
+
+# Copy application code
+COPY --chown=user . /app
+
+# Run the application
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,110 @@
+import os
+import re
+import io
+import zipfile
+import tempfile
+import requests
+from fastapi import FastAPI, HTTPException, Form
+from fastapi.responses import StreamingResponse
+from playwright.async_api import async_playwright
+from bs4 import BeautifulSoup
+
+app = FastAPI()
+
+@app.get("/")
+async def read_root():
+    return {"message": "Hello, World!"}
+
+
+async def login(page, username, password):
+    await page.goto("https://portal.intiaro.com/login?configuratorVersion=2.5")
+    await page.fill('input[name="userName"]', username)
+    await page.fill('input[name="password"]', password)
+    await page.click('button[type=submit]')
+    await page.wait_for_timeout(2000)  # Wait for navigation after login
+
+
+async def siloshot_making(page):
+    await page.wait_for_timeout(5000)  # Wait for the page to fully load
+    html = await page.content()
+    soup = BeautifulSoup(html, 'html.parser')
+    slider = soup.find('img', class_='slider-image')
+    img = slider.get('src')
+    return img
+
+
+def generate_urls(img, start_angle=0, end_angle=360, step=10):
+    angle_pattern = re.compile(r'(angle/)(\d+)')
+    match = angle_pattern.search(img)
+    if match:
+        base_url = img[:match.start(2)]  # Base URL up to the angle value
+
+    urls = []
+    for angle in range(start_angle, end_angle + 1, step):
+        new_url = re.sub(r'angle/\d+', f'angle/{angle}', img)
+        urls.append(new_url)
+    return urls
+
+
+async def download_images(img_urls, product_name):
+    angle_pattern = re.compile(r'angle/(\d+)')
+    count_img_downloaded = 0
+    image_paths = []
+
+    with tempfile.TemporaryDirectory() as tmpdirname:
+        for img_url in img_urls:
+            try:
+                img_response = requests.get(img_url)
+                img_response.raise_for_status()
+                angle_match = angle_pattern.search(img_url)
+                if angle_match:
+                    angle_value = angle_match.group(1)
+                    img_name = f"{product_name}_{angle_value}.png"
+                else:
+                    img_name = "product_unknown.jpg"
+                img_path = os.path.join(tmpdirname, img_name)
+                image_paths.append(img_path)
+                with open(img_path, 'wb') as file:
+                    file.write(img_response.content)
+                count_img_downloaded += 1
+            except Exception as e:
+                print(f'Failed to download {img_url}: {e}')
+
+        zip_buffer = io.BytesIO()
+        with zipfile.ZipFile(zip_buffer, 'w') as zipf:
+            for img_path in image_paths:
+                zipf.write(img_path, os.path.basename(img_path))
+
+        zip_buffer.seek(0)
+        return zip_buffer
+
+
+async def process_images(username, password, product_url, product_name):
+    async with async_playwright() as playwright:
+        browser = await playwright.chromium.launch(headless=True)
+        context = await browser.new_context()
+        page = await context.new_page()
+        await login(page, username, password)
+        await page.goto(product_url)
+        img = await siloshot_making(page)
+        img_urls = generate_urls(img, start_angle=0, end_angle=360, step=10)
+        zip_buffer = await download_images(img_urls, product_name)
+        await browser.close()
+        return zip_buffer
+
+
+@app.post("/generate-images/")
+async def generate_images(username: str = Form(...), password: str = Form(...), product_url: str = Form(...),
+                          product_name: str = Form(...)):
+    if not (username and password and product_url and product_name):
+        raise HTTPException(status_code=400, detail="All input fields are required.")
+
+    zip_buffer = await process_images(username, password, product_url, product_name)
+    return StreamingResponse(zip_buffer, media_type="application/zip",
+                             headers={"Content-Disposition": f"attachment; filename={product_name}_images.zip"})
+
+
+if __name__ == "__main__":
+    import uvicorn
+
+    uvicorn.run(app, host="0.0.0.0", port=7860)

requirements.txt

@@ -0,0 +1,14 @@
+aiofiles==23.2.1
+beautifulsoup4==4.12.3
+fastapi==0.95.2
+packaging==24.1
+pandas==2.2.2
+playwright==1.39.0
+pydantic==1.8.2
+python-multipart==0.0.9
+requests==2.32.3
+urllib3==2.2.2
+uvicorn==0.22.0
+watchfiles==0.22.0
+websockets==12.0
+glob2==0.7