Update app.py

app.py

@@ -449,64 +449,64 @@ 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 + '\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 = "Berhasil Generate Artikel"
-    return status
-    # except:
-    #     gambar = Image.open('error.png')
-    #     return gambar
+    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 = "Berhasil Generate Artikel"
+        return status
+    except:
+        status = "Gagal Generate Artikel"
+        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,14 +549,20 @@ def post(endpoint,endpoint_media,username,password,tags,categories,metode):
     return response_post.json()
 
 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()
-    image = Image.open('output_image.png')
-
-    return title,image,kontent
+    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
 
     
 with gr.Blocks(theme = "soft", title="Wordpress Article Generator") as article_generator:
@@ -574,9 +580,9 @@ with gr.Blocks(theme = "soft", title="Wordpress Article Generator") as article_g
             backlink = gr.Textbox(placeholder="Masukkan Backlink Yang Akan Diterapkan", label="Backlink")
             keyword = gr.Textbox(placeholder="Masukkan Keyword Artikel", label="Keyword")
             versi = gr.Radio(["openai", "azure"], label="Skema Request", info="Pilih Skema Untuk Request ke ChatGPT ")
-            api_key = gr.Textbox(placeholder="Masukkan Api Key", label="API Key")
-            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")
+            api_key = gr.Textbox(placeholder="Masukkan Api Key", type="password",label="API Key")
+            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.Textbox("", label="Generate Status")
             button_scrap.click(fn=scrap, inputs=[link,backlink,keyword,versi,api_key,link_azure,replicate_token], outputs=status)  
@@ -594,8 +600,8 @@ with gr.Blocks(theme = "soft", title="Wordpress Article Generator") as article_g
         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")
-            username= gr.Textbox(placeholder="Masukkan Username Wordpress", label="Username")
-            password= gr.Textbox(placeholder="Masukkan Password Wordpress", label="Password")
+            username= gr.Textbox(placeholder="Masukkan Username Wordpress",label="Username")
+            password= gr.Textbox(placeholder="Masukkan Password Wordpress",type="password" ,label="Password")
             tags= gr.Textbox(placeholder="Masukkan ID Tags Yang Akan Diterapkan", label="Tags")
             categories= gr.Textbox(placeholder="Masukkan ID Kategori Artikel", label="Kategori")
             metode= gr.Radio(["publish", "draft"], label="Status Post", info="Pilih Metode Publish atau Draft Untuk Memposting")