Update app.py

app.py

@@ -202,7 +202,28 @@ def clean_scrap(artikel,link,models,api_key,azure_api_base,keyword):
     contents = content[1:]
     contents = [' '.join(contents).replace("article:", '').replace("Article:", '').strip()]
 
-    return title, judul, link, contents
+    if len(contents)>=2500:
+        return title, judul, link, contents
+    else:
+        with open('log_activity.txt', 'r') as file:
+            existing_data = file.read()
+        
+        log = link + "\n"
+        
+        combined_data = existing_data + log
+        
+        with tempfile.NamedTemporaryFile(mode='w', delete=False) as temp_file:
+            temp_file.write(combined_data)
+         
+        repo_name = get_full_repo_name(model_id="Article_Gen4", token="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")
+        file_url = upload_file(
+            path_or_fileobj=temp_file.name,
+            path_in_repo="log_activity.txt",
+            repo_id=repo_name,
+            repo_type="space",
+            token="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",
+        )
+        return None,None,None,None
 
 def scrap_artikel(source_type,source,models,api_key,azure_api_base,keyword):
     options = webdriver.ChromeOptions()
@@ -517,17 +538,7 @@ def scrap(source_type,source,backlink,keyword,version,api_key,azure_api_base,rep
         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 = url + "\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)
         
@@ -535,14 +546,6 @@ def scrap(source_type,source,backlink,keyword,version,api_key,azure_api_base,rep
             file.write(kontent)
         
         repo_name = get_full_repo_name(model_id="Article_Gen4", token="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")
-        file_url = upload_file(
-            path_or_fileobj=temp_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",
@@ -565,19 +568,17 @@ def scrap(source_type,source,backlink,keyword,version,api_key,azure_api_base,rep
             token="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",
         )
         if kontent:
-            status = "<h3>Berhasil Generate Artikel</h3>"
-            time.sleep(60)
-            return status,gambar
-        else:
             with open('log_activity.txt', 'r') as file:
                 existing_data = file.read()
-    
-            log = source + "\n"
+            
+            log = url + "\n"
+            
             combined_data = existing_data + log
-    
+            
             with tempfile.NamedTemporaryFile(mode='w', delete=False) as temp_file:
                 temp_file.write(combined_data)
-
+            
+            
             repo_name = get_full_repo_name(model_id="Article_Gen4", token="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")
             file_url = upload_file(
                 path_or_fileobj=temp_file.name,
@@ -586,28 +587,14 @@ def scrap(source_type,source,backlink,keyword,version,api_key,azure_api_base,rep
                 repo_type="space",
                 token="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",
             )
+            status = "<h3>Berhasil Generate Artikel</h3>"
+            time.sleep(60)
+            return status,gambar
+        else:
             status = "<h3>Gagal Generate Artikel, Coba Generate Ulang atau Ganti Input Source</h3>"
             gambar = Image.open("error.png")
             return status,gambar
     except:
-        with open('log_activity.txt', 'r') as file:
-            existing_data = file.read()
-
-        log = source + "\n"
-        combined_data = existing_data + log
-
-        with tempfile.NamedTemporaryFile(mode='w', delete=False) as temp_file:
-            temp_file.write(combined_data)
-            
-        repo_name = get_full_repo_name(model_id="Article_Gen4", token="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")
-        file_url = upload_file(
-            path_or_fileobj=temp_file.name,
-            path_in_repo="log_activity.txt",
-            repo_id=repo_name,
-            repo_type="space",
-            token="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",
-        )
-        
         status = "<h3>Gagal Generate Artikel</h3>"
         gambar = Image.open('error.png')
         return status,gambar