Update app.py

app.py

@@ -486,9 +486,9 @@ def scrap(link_scrap,backlink,keyword,version,api_key,azure_api_base,replicate_k
         repo_type="space",
         token="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",
     )
-    # if kontent:
-    #     status = "Berhasil Generate Artikel"
-    return gambar
+    if kontent:
+        status = "Berhasil Generate Artikel"
+    return status
     # except:
     #     gambar = Image.open('error.png')
     #     return gambar
@@ -533,20 +533,16 @@ def post(endpoint,endpoint_media,username,password,tags,categories,metode):
     response_post = requests.post(endpoint, headers=headers, data=data)
     return response_post.json()
 
-def html_view():
+def view_outputs():
     with open('judul.txt', 'r') as file:
         judul = file.read()
         title = '<h1>'+judul+'</h1>'
     with open('kontent.txt', 'r') as file:
         kontent = file.read()
-    return title,kontent
+    with open('image_data.txt', 'rb') as file:
+        file_content = file.read()
+    return title,file_content,kontent
 
-def raw_text_view():
-    with open('judul.txt', 'r') as file:
-        judul = file.read()
-    with open('kontent.txt', 'r') as file:
-        kontent = file.read()
-    return judul,kontent
     
 with gr.Blocks(theme = "soft", title="Wordpress Article Generator") as article_generator:
     gr.HTML(
@@ -567,18 +563,18 @@ with gr.Blocks(theme = "soft", title="Wordpress Article Generator") as article_g
             link_azure = gr.Textbox(placeholder="Khusus Untuk Skema Request Menggunakan Azure", label="Azure Endpoint (Opsional)")
             replicate_token = gr.Textbox(placeholder="Masukkan Token Replicate", label="Replicate Key")
             button_scrap = gr.Button("Scrap Article")
-            img = gr.Image()
-            button_scrap.click(fn=scrap, inputs=[link,backlink,keyword,versi,api_key,link_azure,replicate_token], outputs=img)    
+            button_scrap.click(fn=scrap, inputs=[link,backlink,keyword,versi,api_key,link_azure,replicate_token], outputs=img)  
+            view_output = gr.Button("View Output")
             with gr.Tab("View_HTML"):
-                html_button = gr.Button("View Output HTML")
                 title = gr.HTML("")
+                img = gr.Image()
                 content = gr.HTML("")
-                html_button.click(fn=html_view, outputs = [title,content])
+                view_output.click(fn=view_outputs, outputs = [title,img,content])
             with gr.Tab("View_Raw_Text"):
-                raw_button = gr.Button("View Raw Text")
                 title = gr.Textbox("", label="Judul")
+                img = gr.Image()
                 content = gr.Textbox("", label="Konten")
-                raw_button.click(fn=raw_text_view, outputs = [title,content])
+                view_output.click(fn=view_outputs, outputs = [title,img,content])
         with gr.Column():
             endpoint= gr.Textbox(placeholder="Masukkan Endpoint Wordpress", label="Endpoint Wordpress")
             endpoint_media= gr.Textbox(placeholder="Masukkan Endpoint Media Wordpress", label="Endpoint Media")