Update app.py

app.py

@@ -449,65 +449,65 @@ def artikel_processing(link_scrap,backlink,keyword,models,api_key,azure_api_base
     return judul,content,image,image_data
 
 def scrap(link_scrap,backlink,keyword,version,api_key,azure_api_base,replicate_key):
-    # try:
-    judul,kontent,gambar, image_data= artikel_processing(link_scrap,backlink,keyword,version,api_key,azure_api_base,replicate_key)
-    title = '<h1>'+judul+'</h1>'
-    desired_timezone = pytz.timezone('Asia/Jakarta')
-    current_time = datetime.datetime.now(desired_timezone)
-    Timestamp = current_time.strftime('%Y-%m-%d %H:%M:%S')
-    
-    with open('log_activity.txt', 'r') as file:
-        existing_data = file.read()
-    
-    log = 'Generated_Title:' + judul + '\nTimestamp:' + Timestamp + '\nLink:' + link_scrap + "\n\n"
-    
-    combined_data = existing_data + log
-    
-    with tempfile.NamedTemporaryFile(mode='w', delete=False) as temp_file:
-        temp_file.write(combined_data)
-    
-    with open("judul.txt", "w") as file:
-        file.write(judul)
-    
-    with open("kontent.txt", "w") as file:
-        file.write(kontent)
-    
-    repo_name = get_full_repo_name(model_id="Article_Gen3", token="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")
-    file_url = upload_file(
-        path_or_fileobj=temp_file.name,  # Use the temporary file name
-        path_in_repo="log_activity.txt",
-        repo_id=repo_name,
-        repo_type="space",
-        token="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",
-    )
-    file_url = upload_file(
-        path_or_fileobj=image_data,
-        path_in_repo="image_data.txt",
-        repo_id=repo_name,
-        repo_type="space",
-        token="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",
-    )
-    file_url = upload_file(
-        path_or_fileobj='judul.txt',
-        path_in_repo="judul.txt",
-        repo_id=repo_name,
-        repo_type="space",
-        token="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",
-    )
-    file_url = upload_file(
-        path_or_fileobj='kontent.txt',
-        path_in_repo="kontent.txt",
-        repo_id=repo_name,
-        repo_type="space",
-        token="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",
-    )
-    
-    return title,image,kontent
-    # except:
-    #     title = 'Gagal Generate Judul'
-    #     kontent = 'Gagal Generate Konten'
-    #     image = Image.open('error.png')
-    #     return title,image,kontent
+    try:
+        judul,kontent,gambar, image_data= artikel_processing(link_scrap,backlink,keyword,version,api_key,azure_api_base,replicate_key)
+        title = '<h1>'+judul+'</h1>'
+        desired_timezone = pytz.timezone('Asia/Jakarta')
+        current_time = datetime.datetime.now(desired_timezone)
+        Timestamp = current_time.strftime('%Y-%m-%d %H:%M:%S')
+        
+        with open('log_activity.txt', 'r') as file:
+            existing_data = file.read()
+        
+        log = 'Generated_Title:' + judul + '\nTimestamp:' + Timestamp + '\nLink:' + link_scrap + "\n\n"
+        
+        combined_data = existing_data + log
+        
+        with tempfile.NamedTemporaryFile(mode='w', delete=False) as temp_file:
+            temp_file.write(combined_data)
+        
+        with open("judul.txt", "w") as file:
+            file.write(judul)
+        
+        with open("kontent.txt", "w") as file:
+            file.write(kontent)
+        
+        repo_name = get_full_repo_name(model_id="Article_Gen3", token="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")
+        file_url = upload_file(
+            path_or_fileobj=temp_file.name,  # Use the temporary file name
+            path_in_repo="log_activity.txt",
+            repo_id=repo_name,
+            repo_type="space",
+            token="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",
+        )
+        file_url = upload_file(
+            path_or_fileobj=image_data,
+            path_in_repo="image_data.txt",
+            repo_id=repo_name,
+            repo_type="space",
+            token="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",
+        )
+        file_url = upload_file(
+            path_or_fileobj='judul.txt',
+            path_in_repo="judul.txt",
+            repo_id=repo_name,
+            repo_type="space",
+            token="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",
+        )
+        file_url = upload_file(
+            path_or_fileobj='kontent.txt',
+            path_in_repo="kontent.txt",
+            repo_id=repo_name,
+            repo_type="space",
+            token="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",
+        )
+        if kontent:
+            status = "<h3>Berhasil Generate Artikel</h3>"
+            return status
+    except:
+        status = "<h3>Gagal Generate Artikel</h3>"
+        return status
+
 
 def post(endpoint,endpoint_media,username,password,tags,categories,metode):
     credentials = base64.b64encode(f"{username}:{password}".encode("utf-8")).decode("utf-8")
@@ -549,6 +549,20 @@ def post(endpoint,endpoint_media,username,password,tags,categories,metode):
     response_post = requests.post(endpoint, headers=headers, data=data)
     return response_post.json()
 
+def view_output():
+    try:
+        with open('judul.txt', 'r') as file:
+            judul = file.read()
+            title = '<h1>'+judul+'</h1>'
+        with open('kontent.txt', 'r') as file:
+            kontent = file.read()
+            image = Image.open('output_image.png')
+        return title,image,kontent
+    except:
+        title = 'Gagal Generate Judul'
+        kontent = 'Gagal Generate Konten'
+        image = Image.open('error.png')
+        return title,image,kontent
         
 def save(title,content):
     with open("judul.txt", "w") as file:
@@ -604,11 +618,14 @@ with gr.Blocks(theme = "soft", title="Wordpress Article Generator") as article_g
             link_azure = gr.Textbox(placeholder="Khusus Untuk Skema Request Menggunakan Azure",type="password", label="Azure Endpoint (Opsional)")
             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)
+            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)
-                button_scrap.click(fn=scrap, inputs=[link,backlink,keyword,versi,api_key,link_azure,replicate_token], outputs=[title,img,content])
+                view_outputs.click(fn=view_output, outputs=[title,img,content])
                 save_button= gr.Button("Save Change")
                 status = gr.HTML("")
                 save_button.click(fn=save, inputs =[title,content], outputs = status)
@@ -616,10 +633,9 @@ with gr.Blocks(theme = "soft", title="Wordpress Article Generator") as article_g
                 title = gr.HTML("")
                 img = gr.Image()
                 content = gr.HTML("")
-                button_scrap.click(fn=scrap, inputs=[link,backlink,keyword,versi,api_key,link_azure,replicate_token], outputs=[title,img,content])
+                view_outputs.click(fn=view_output, outputs=[title,img,content])
                 view_change = gr.Button("View Change")
                 view_change.click(fn=view, outputs=[title,content])
-            # button_scrap.click(fn=scrap, inputs=[link,backlink,keyword,versi,api_key,link_azure,replicate_token], 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")