Update app.py

app.py

@@ -400,11 +400,12 @@ def get_image(prompt):
         sdxl_version = sdxl_model.versions.get("a00d0b7dcbb9c3fbb34ba87d2d5b46c56969c84a628bf778a7fdaec30b1b99c5")
     
         prediction = replicate.predictions.create(version=sdxl_version,
-            input={"prompt": 'Phantasmal iridescent, vibrant color, high contrast, award winning, trending in artstation, digital art' + prompt,
-                    "negative_prompt": 'NSFW, cityscape',
+            input={"prompt":'Phantasmal iridescent, vibrant color, high contrast, award winning, trending in artstation, digital art' + prompt,
+                    "negative_prompt":'NSFW, cityscape',
                     "width": 1648,
-                    "height": 1024}
+                    "height":1024}
             )  
+        prediction
         prediction.reload()
         prediction.wait()
     
@@ -412,33 +413,14 @@ def get_image(prompt):
             print(f"Error: {prediction.error}")
             print(f"Last Log")
             print(prediction.logs)
-            return None
-        elif (prediction.status == 'succeeded' and prediction.output is not None):
+        elif ((prediction.status == 'succeeded') | (prediction.output != None)):
             response = requests.get(prediction.output[0])
-            image_base64 = base64.b64encode(response.content).decode('utf-8')
-            return image_base64
-    except Exception as e:
-        print(f"An error occurred: {e}")
-        return None
+            image_data = base64.b64encode(response.content)
+            return image_data
+    except:
+        image_data = None
+        return image_data
         
-def show_image(image_data, width=800, height=600):
-    # Konversi data gambar base64 menjadi gambar PIL
-    image_bytes = base64.b64decode(image_data)
-    image = Image.open(io.BytesIO(image_bytes))
-    
-    # Resize gambar sesuai dengan lebar dan tinggi yang diinginkan
-    image = image.resize((width, height), Image.ANTIALIAS)
-    
-    # Konversi kembali gambar ke format base64
-    buffered = io.BytesIO()
-    image.save(buffered, format="JPEG")
-    image_data_resized = base64.b64encode(buffered.getvalue()).decode('utf-8')
-    
-    return gr.outputs.Image(image_data_resized)
-
-iface = gr.Interface(fn=get_image, inputs="text", outputs=show_image)
-iface.launch()
-
 #     return tags, content, url #image, image_data
 
 # def scraping_article(link_scrap):
@@ -447,21 +429,21 @@ iface.launch()
 #     return judul, content
     
 
-# with gr.Blocks(theme = "soft", title="Article Writer") as article_generator:
-#     gr.Markdown(
-#             """
-#             # Article Writer
-#             Buat artikel yang anda inginkan dengan mudah, cukup menuliskan keyword atau topik artikel yang ingin anda buat dan dalam sekejap artikel tersebut menjadi milik anda.
-#             """)
-#     with gr.Row():
-#         inputs = gr.Textbox(placeholder="Link Article", show_label=False)
-#     with gr.Row():
-#         button_scrap = gr.Button("Scrap Article")
-#     with gr.Row():
-#         # title =  gr.Textbox("", label="Title")
-#         # content =  gr.Textbox("", label="Content")
-#         img_data = gr.Textbox("", label="Content")
-#         img = gr.Image("", label="Image")
-#         button_scrap.click(fn=get_image, inputs=inputs, outputs=[img, img_data])
-# if __name__ == "__main__":
-#     article_generator.launch(share=False)
+with gr.Blocks(theme = "soft", title="Article Writer") as article_generator:
+    gr.Markdown(
+            """
+            # Article Writer
+            Buat artikel yang anda inginkan dengan mudah, cukup menuliskan keyword atau topik artikel yang ingin anda buat dan dalam sekejap artikel tersebut menjadi milik anda.
+            """)
+    with gr.Row():
+        inputs = gr.Textbox(placeholder="Link Article", show_label=False)
+    with gr.Row():
+        button_scrap = gr.Button("Scrap Article")
+    with gr.Row():
+        # title =  gr.Textbox("", label="Title")
+        # content =  gr.Textbox("", label="Content")
+        # img_data = gr.Textbox("", label="Content")
+        img = gr.Image("", label="Image")
+        button_scrap.click(fn=get_image, inputs=inputs, outputs=img)
+if __name__ == "__main__":
+    article_generator.launch(share=False)