Update app.py

app.py

@@ -405,7 +405,6 @@ def get_image(prompt):
                     "width": 1648,
                     "height": 1024}
             )  
-        prediction
         prediction.reload()
         prediction.wait()
     
@@ -413,21 +412,32 @@ def get_image(prompt):
             print(f"Error: {prediction.error}")
             print(f"Last Log")
             print(prediction.logs)
-        elif (prediction.status == 'succeeded' or prediction.output is not None):
+            return None
+        elif (prediction.status == 'succeeded' and prediction.output is not None):
             response = requests.get(prediction.output[0])
-            image_base64 = base64.b64encode(response.content)
-            image = Image.open(io.BytesIO(base64.b64decode(image_base64)))
-            image = image.crop((3, 0, 1645, 1024))
-            w, h = image.size
-            new_w = int(w / 1.641)
-            new_h = int(h / 1.641)
-            image = image.resize((new_w, new_h), Image.ANTIALIAS)
-            image_data = base64.b64encode(image.tobytes()).decode('utf-8')
-            return image_data, image
+            image_base64 = base64.b64encode(response.content).decode('utf-8')
+            return image_base64
     except Exception as e:
         print(f"An error occurred: {e}")
         return None
+        
+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
 
@@ -437,21 +447,21 @@ def get_image(prompt):
 #     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, img_data])
+# if __name__ == "__main__":
+#     article_generator.launch(share=False)