Update app.py

app.py

@@ -97,43 +97,75 @@ def upload_image(url):
         print(f"Failed to upload: {url}")
         return None
 
-@app.post("/generate-image")
-async def generate_image(request: ImageRequest):
-    print("Image Generation Request", request)
-    try:
-        runware = Runware(api_key=RUNWARE_API_KEY)
-        await runware.connect()
+# @app.post("/generate-image")
+# async def generate_image(request: ImageRequest):
+#     print("Image Generation Request", request)
+#     try:
+#         runware = Runware(api_key=RUNWARE_API_KEY)
+#         await runware.connect()
 
-        request_image = IImageInference(
-            positivePrompt=request.prompt,
-            model=request.model,
-            numberResults=request.number_results,
-            height=request.height,
-            width=request.width,
-        )
+#         request_image = IImageInference(
+#             positivePrompt=request.prompt,
+#             model=request.model,
+#             numberResults=request.number_results,
+#             height=request.height,
+#             width=request.width,
+#         )
 
-        images = await runware.imageInference(requestImage=request_image)
-        image_urls = [image.imageURL for image in images]
-        print("Generated Images: ", image_urls)
-        # imgbb_urls = []
+#         images = await runware.imageInference(requestImage=request_image)
+#         image_urls = [image.imageURL for image in images]
+#         print("Generated Images: ", image_urls)
+#         # imgbb_urls = []
 
-        # with ThreadPoolExecutor(max_workers=10) as executor:
-        #     future_to_url = {executor.submit(upload_image, url): url for url in image_urls}
-        #     for future in as_completed(future_to_url):
-        #         url = future_to_url[future]
-        #         try:
-        #             imgbb_url = future.result()
-        #             if imgbb_url:
-        #                 imgbb_urls.append(imgbb_url)
-        #         except Exception as exc:
-        #             print(f"{url} generated an exception: {exc}")
+#         # with ThreadPoolExecutor(max_workers=10) as executor:
+#         #     future_to_url = {executor.submit(upload_image, url): url for url in image_urls}
+#         #     for future in as_completed(future_to_url):
+#         #         url = future_to_url[future]
+#         #         try:
+#         #             imgbb_url = future.result()
+#         #             if imgbb_url:
+#         #                 imgbb_urls.append(imgbb_url)
+#         #         except Exception as exc:
+#         #             print(f"{url} generated an exception: {exc}")
 
-        batch_id = insert_batch(request.prompt, request.width, request.height, request.model, image_urls)
+#         batch_id = insert_batch(request.prompt, request.width, request.height, request.model, image_urls)
         
-        response = {"batch": {"id": batch_id, "prompt": request.prompt, "width": request.width, "height": request.height, "model": request.model, "images": [{"url": url} for url in image_urls]}}
-        return response
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Failed to generate image: {str(e)}")
+#         response = {"batch": {"id": batch_id, "prompt": request.prompt, "width": request.width, "height": request.height, "model": request.model, "images": [{"url": url} for url in image_urls]}}
+#         return response
+#     except Exception as e:
+#         raise HTTPException(status_code=500, detail=f"Failed to generate image: {str(e)}")
+
+@app.post("/generate-image")
+ async def generate_image(request: ImageRequest):
+     print("Image Generation Request", request)
+     try:
+         runware = Runware(api_key=RUNWARE_API_KEY)
+         await runware.connect()
+
+         request_image = IImageInference(
+             positivePrompt=request.prompt,
+             model=request.model,
+             numberResults=request.number_results,
+             height=request.height,
+             width=request.width,
+         )
+
+         images = await runware.imageInference(requestImage=request_image)
+         image_urls = [image.imageURL for image in images]
+         print("Generated Images: ", image_urls)
+
+         response = {
+             "batch": {
+                 "prompt": request.prompt,
+                 "width": request.width,
+                 "height": request.height,
+                 "model": request.model,
+                 "images": [{"url": url} for url in image_urls]
+             }
+         }
+         return response
+     except Exception as e:
+         raise HTTPException(status_code=500, detail=f"Failed to generate image: {str(e)}")
 
 @app.get("/get-batches")
 async def get_batches():