Update app.py

app.py

@@ -31,6 +31,35 @@ def split_article(article_text):
 
     return first_quarter, second_quarter, third_quarter, fourth_quarter
 
+def replace_content(content, replacements):
+    for pattern, replacement in replacements.items():
+        content = re.sub(pattern, replacement, content)
+    return content
+
+def generate_patterns(base_replacements):
+    patterns = {}
+    for key, value in base_replacements.items():
+        patterns[key] = value
+        patterns[key.capitalize()] = value.capitalize()
+        patterns[key.upper()] = value.upper()
+        patterns[key.lower()] = value.lower()
+    return patterns
+
+base_replacements = {
+    'Layanan Pelanggan': 'Customer Service',
+    'Pusat Kontak': 'Contact Center',
+    'Multi Kanal': 'Omnichannel',
+    'Saluran Omni': 'Omnichannel',
+    'Merek':'Brand',
+    'Komputasi Awan':'Cloud Computing',
+    'Kecerdasan Buatan':'Artificial Intelligence',
+    'Pembelajaran Mesin':'Machine Learning',
+    'Alat Layanan Pelanggan':'Customer Service Tools',
+    'Pengalaman Pelanggan':'Customer Experience',
+    'AI Percakapan':'AI Conversation',
+    'Aplikasi pesan':'Message app'
+}
+
 def clean_scrap(artikel,link):
     new_artikel = []
     article = []
@@ -91,6 +120,11 @@ def clean_scrap(artikel,link):
     judul = response['choices'][0]['message']['content']
     judul = judul.replace("Judul:", '').strip()
     judul = judul.replace("Title:", '').strip()
+    try:
+        replacements = generate_patterns(base_replacements)
+        judul = replace_content(judul, replacements)
+    except:
+        judul = judul
     contents = content[1:]
     contents = [' '.join(contents).replace("article:", '').replace("Article:", '').strip()]
 
@@ -131,34 +165,6 @@ def scrap_artikel(link_scrap):
     title, judul, url, contents = clean_scrap(artikels,link_scrap)
     return title, judul, url, contents
 
-def replace_content(content, replacements):
-    for pattern, replacement in replacements.items():
-        content = re.sub(pattern, replacement, content)
-    return content
-
-def generate_patterns(base_replacements):
-    patterns = {}
-    for key, value in base_replacements.items():
-        patterns[key] = value
-        patterns[key.capitalize()] = value.capitalize()
-        patterns[key.upper()] = value.upper()
-        patterns[key.lower()] = value.lower()
-    return patterns
-
-base_replacements = {
-    'Layanan Pelanggan': 'Customer Service',
-    'Pusat Kontak': 'Contact Center',
-    'Multi Kanal': 'Omnichannel',
-    'Saluran Omni': 'Omnichannel',
-    'Merek':'Brand',
-    'Komputasi Awan':'Cloud Computing',
-    'Kecerdasan Buatan':'Artificial Intelligence',
-    'Pembelajaran Mesin':'Machine Learning',
-    'Alat Layanan Pelanggan':'Customer Service Tools',
-    'Pengalaman Pelanggan':'Customer Experience',
-    'AI Percakapan':'AI Conversation',
-    'Aplikasi pesan':'Message app'
-}
 def artikel_processing(link_scrap,backlink,keyword):
     title, judul, url, artikel= scrap_artikel(link_scrap)
     teks_to_tags = artikel[0][:500]
@@ -473,7 +479,7 @@ with gr.Blocks(theme = "soft", title="Article Writer") as article_generator:
     gr.Markdown(
             """
             # Wordpress Article Generator
-            Buat artikel yang anda inginkan dengan mudah, cukup menuliskan keyword atau topik artikel yang ingin anda buat dan dalam sekejap artikel tersebut menjadi milik anda.
+            Buat artikel yang anda inginkan dengan mudah, cukup dengan menuliskan link artikel sebagai referensi dan dalam sekejap artikel yang anda inginkan dapat menghiasi halaman wordpress anda.
             """)
     with gr.Row():
         with gr.Column():