Update app.py

app.py

@@ -27,6 +27,7 @@ from huggingface_hub import (
     upload_file,
 )
 logs = []
+image_datas = []
 
 def split_article(article_text):
     words = article_text.split()    
@@ -422,12 +423,10 @@ def artikel_processing(link_scrap,backlink,keyword,openai_key,replicate_key):
             os.remove(tmp_path)
     except:
         image_data = None
+    image_datas.append(image_data)
     data = [{
         'Judul Artikel':judul,
-        'Konten Artikel':content,
-        'Image':image if 'image' in locals() else None,
-        'image_data':image_data
-        
+        'Konten Artikel':content
     }]
     result = pd.DataFrame(data)
     return result,image
@@ -465,19 +464,16 @@ def scrap(link_scrap,backlink,keyword,openai_key,replicate_key):
     #     kontent = 'Gagal Generate Artikel:Request Time Out'
     #     return gambar,judul,kontent
 
-def post(judul, kontent, gambar, endpoint,endpoint_media,username,password,tags,categories,metode):
+def post(judul, kontent, endpoint,endpoint_media,username,password,tags,categories,metode):
     credentials = base64.b64encode(f"{username}:{password}".encode("utf-8")).decode("utf-8")
     headers = {"Authorization": f"Basic {credentials}"}
 
-    gambar_bytes = gambar.tobytes()
-    gambar_base64 = base64.b64encode(gambar_bytes).decode('utf-8')
-
     data = {
         "alt_text": judul,
         "media_type": "image",
         "mime_type": "png"
     }
-    files = {"file": ("image.jpg", gambar_base64)} 
+    files = {"file": ("image.jpg", image_datas[0])}
     try :
         response_media= requests.post(endpoint_media, headers=headers, data=data, files=files) # Send
 
@@ -527,6 +523,6 @@ with gr.Blocks(theme = "soft", title="Wordpress Article Generator") as article_g
             metode= gr.Radio(["publish", "draft"], label="Status Post", info="Pilih Metode Publish atau Draft Untuk Memposting")
             button_post = gr.Button("Post Article")
             status = gr.Textbox("", label="Response")
-            button_post.click(fn=post, inputs=[outputs, content, img, endpoint,endpoint_media,username,password,tags,categories,metode], outputs=status)
+            button_post.click(fn=post, inputs=[outputs, content, endpoint,endpoint_media,username,password,tags,categories,metode], outputs=status)
 if __name__ == "__main__":
     article_generator.launch(share=False)