Update app.py

app.py

@@ -415,6 +415,21 @@ def artikel_processing(link_scrap,backlink,keyword,models,api_key,azure_api_base
                 base64_string = base64.b64decode(base64_string)
             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)
+        
+        # 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')
         image = image.crop((3,0,1645,1024))
@@ -539,9 +554,9 @@ def view_outputs():
         title = '<h1>'+judul+'</h1>'
     with open('kontent.txt', 'r') as file:
         kontent = file.read()
-    with open('image_data.txt', 'rb') as file:
-        file_content = file.read()
-    return title,file_content,kontent
+    image = Image.open('output_image.png')
+
+    return title,image,kontent
 
     
 with gr.Blocks(theme = "soft", title="Wordpress Article Generator") as article_generator: