Update app.py

app.py

@@ -97,64 +97,40 @@ 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()
-
-#         request_image = IImageInference(
-#             positivePrompt=request.prompt,
-#             model=request.model,
-#             numberResults=request.number_results,
-#             height=request.height,
-#             width=request.width,
-#         )
+@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()
 
-#         images = await runware.imageInference(requestImage=request_image)
-#         image_urls = [image.imageURL for image in images]
-#         print("Generated Images: ", image_urls)
-#         # imgbb_urls = []
+        request_image = IImageInference(
+            positivePrompt=request.prompt,
+            model=request.model,
+            numberResults=request.number_results,
+            height=request.height,
+            width=request.width,
+        )
 
-#         # 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}")
+        images = await runware.imageInference(requestImage=request_image)
+        image_urls = [image.imageURL for image in images]
+        print("Generated Images: ", image_urls)
+        # imgbb_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)}")
+        # 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}")
 
-@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,
-             )
+        # batch_id = insert_batch(request.prompt, request.width, request.height, request.model, image_urls)
         
-            images = await runware.imageInference(requestImage=request_image)
-            image_urls = [image.imageURL for image in images]
-            print("Generated Images: ", image_urls)
-        
-            response = {
+        response = {
                  "batch": {
                      "prompt": request.prompt,
                      "width": request.width,
@@ -163,9 +139,41 @@ def upload_image(url):
                      "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)}")
+        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():