Update app.py

app.py

@@ -417,19 +417,19 @@ def artikel_processing(link_scrap,backlink,keyword,models,api_key,azure_api_base
             image_data= base64_string
             os.remove(tmp_path)
             
-        output_image_io = io.BytesIO()
-        image.save(output_image_io, format='PNG')
-        output_image_io.seek(0)
+        # output_image_io = io.BytesIO()
+        # image.save(output_image_io, format='PNG')
+        # output_image_io.seek(0)
         
-        # Upload gambar ke repo Hugging Face
-        repo_name = get_full_repo_name(model_id="Article_Gen3", token="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")
-        file_url = upload_file(
-            path_or_fileobj=output_image_io.read(),
-            path_in_repo="output_image.png",
-            repo_id=repo_name,
-            repo_type="space",
-            token="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",
-        )
+        # # Upload gambar ke repo Hugging Face
+        # repo_name = get_full_repo_name(model_id="Article_Gen3", token="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")
+        # file_url = upload_file(
+        #     path_or_fileobj=output_image_io.read(),
+        #     path_in_repo="output_image.png",
+        #     repo_id=repo_name,
+        #     repo_type="space",
+        #     token="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",
+        # )
         
     except:
         image = Image.open('botika_logo.jpeg')
@@ -505,9 +505,10 @@ def scrap(link_scrap,backlink,keyword,version,api_key,azure_api_base,replicate_k
         if kontent:
             status = "<h3>Berhasil Generate Artikel</h3>"
             time.sleep(60)
-            return status
+            return status,gambar
     except:
         status = "<h3>Gagal Generate Artikel</h3>"
+        gambar = Image.open('error.png')
         return status
 
 
@@ -622,22 +623,21 @@ with gr.Blocks(theme = "soft", title="Wordpress Article Generator") as article_g
             replicate_token = gr.Textbox(placeholder="Masukkan Token Replicate", type="password",label="Replicate Key")
             button_scrap = gr.Button("Scrap Article")
             status = gr.HTML("")
-            button_scrap.click(fn=scrap, inputs=[link,backlink,keyword,versi,api_key,link_azure,replicate_token], outputs= status)
+            gambar = gr.Image(label="Generate Gambar")
+            button_scrap.click(fn=scrap, inputs=[link,backlink,keyword,versi,api_key,link_azure,replicate_token], outputs= status,gambar)
             view_outputs = gr.Button("View Output")
             with gr.Tab("Raw Output"):
                 title = gr.Textbox("", label="Judul", interactive=True)
-                img = gr.Image(height=300, width=500)
                 content = gr.Textbox("", label="Konten", interactive=True)
-                view_outputs.click(fn=view_output, outputs=[title,img,content])
+                view_outputs.click(fn=view_output, outputs=[title,content])
                 save_button= gr.Button("Save Change")
                 status = gr.HTML("")
                 save_button.click(fn=save, inputs =[title,content], outputs = status)
             with gr.Tab("Formatted Output"):
                 view_change = gr.Button("View Change")
                 title = gr.HTML("")
-                img = gr.Image()
                 content = gr.HTML("")
-                view_outputs.click(fn=view_output, outputs=[title,img,content])
+                view_outputs.click(fn=view_output, outputs=[title,content])
                 view_change.click(fn=view, outputs=[title,content])
         with gr.Column():
             endpoint= gr.Textbox(placeholder="Masukkan Endpoint Wordpress", label="Endpoint Wordpress")