Update app.py

app.py

@@ -7,16 +7,18 @@ from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
 from deep_translator import GoogleTranslator
 import yake
 from datetime import datetime
+import shutil
+import glob
 
 # ✅ Set API Key
 # os.environ["GROQ_API_KEY"] = "your_api_key_here"
 # client = Groq(api_key=os.environ["GROQ_API_KEY"])
 
 # ✅ Set API Key
-os.environ["GROQ_API_KEY"] = "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"    # Replace with your API key
+os.environ["GROQ_API_KEY"] = "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"    # Replace with your API key
 client = Groq(api_key=os.environ["GROQ_API_KEY"])
 
-# ✅ Load M2M-100 Model
+# # ✅ Load M2M-100 Model
 # model_name = "facebook/m2m100_418M"
 # tokenizer = M2M100Tokenizer.from_pretrained(model_name)
 # model = M2M100ForConditionalGeneration.from_pretrained(model_name)
@@ -43,15 +45,6 @@ def generate_script(topic, duration):
     except Exception as e:
         return f"❌ Error in script generation: {str(e)}"
 
-# # ✅ Function to Extract Keywords
-# def extract_keywords(script):
-#     words = word_tokenize(script.lower())
-#     stop_words = set(stopwords.words('english'))
-#     filtered_words = [word for word in words if word.isalnum() and word not in stop_words]
-#     word_freq = Counter(filtered_words)
-#     keywords = [word for word, freq in word_freq.most_common(10)]
-#     return ", ".join(keywords)
-
 
 # ✅ Function to Extract Keywords Using YAKE
 def extract_keywords(script):
@@ -69,22 +62,6 @@ def extract_keywords(script):
         return f"❌ Error extracting keywords: {str(e)}"
 
 
-# # ✅ Function to Save Edited or Generated Keywords
-# def save_keywords_file(keywords):
-#     filename = "keywords.txt"
-#     with open(filename, "w", encoding="utf-8") as f:
-#         f.write(keywords)
-#     return filename
-
-# def save_keywords_file(keywords):
-#     if not keywords.strip():
-#         return "❌ No keywords available to save."
-    
-#     filename = "keywords.txt"
-#     with open(filename, "w", encoding="utf-8") as f:
-#         f.write(keywords)
-#     return filename
-
 def save_keywords_file(keywords, topic):
     today = datetime.today().strftime('%Y_%b_%d')
     filename = f"Keywords/{topic}_Keyword_{today}.txt"
@@ -120,17 +97,20 @@ def save_keywords_file(keywords, topic):
 def translate_to_urdu(english_script):
     try:
         # ✅ Define a max chunk size (Google Translator has a limit)
-        max_chunk_size = 5000
+        max_chunk_size = 4500  # Stay below 5000 to be safe
         chunks = [english_script[i:i + max_chunk_size] for i in range(0, len(english_script), max_chunk_size)]
 
-        # ✅ Translate each chunk separately
-        translated_chunks = [GoogleTranslator(source='en', target='ur').translate(chunk) for chunk in chunks]
+        translated_chunks = []
+        for chunk in chunks:
+            translated_chunk = GoogleTranslator(source='en', target='ur').translate(chunk)
+            translated_chunks.append(translated_chunk)
 
         return " ".join(translated_chunks)  # ✅ Join all translated chunks
     except Exception as e:
         return f"❌ Error in translation: {str(e)}"
 
 
+
 def save_english_file(content, topic):
     today = datetime.today().strftime('%Y_%b_%d')  # Format: 2025_Feb_21
     filename = f"English_Scripts/{topic}_Eng_{today}.txt"
@@ -162,6 +142,22 @@ def finalize_process():
     return "✅ Script Generation Completed Successfully!"
 
 
+def clear_old_files():
+    # ✅ Define all directories where files are stored
+    directories = ["English_Scripts", "Urdu_Scripts", "Urdu_Final", "Keywords"]
+
+    for directory in directories:
+        if os.path.exists(directory):  # ✅ Check if directory exists
+            files = glob.glob(f"{directory}/*")  # ✅ Get all files in the directory
+            for file in files:
+                try:
+                    os.remove(file)  # ✅ Delete each file
+                except Exception as e:
+                    print(f"❌ Error deleting {file}: {e}")
+
+    return "", "", "", "", ""  # ✅ Clear all textboxes in UI
+
+
 
 
 # ✅ Gradio UI
@@ -182,7 +178,7 @@ with gr.Blocks() as app:
     extract_keywords_btn = gr.Button("🔑 Extract Keywords")
     keyword_output = gr.Textbox(label="🔍 Extracted Keywords", interactive=True)
     download_keywords_btn = gr.Button("⬇️ Download Keywords")
-    download_keywords_btn.click(save_keywords_file, inputs=[keyword_output], outputs=[gr.File()])
+    download_keywords_btn.click(save_keywords_file, inputs=[keyword_output, topic_input], outputs=[gr.File()])
 
     translate_button = gr.Button("Generate Urdu Script")
     urdu_output = gr.Textbox(label="Translated Urdu Script", interactive=False, rtl=True)
@@ -196,7 +192,8 @@ with gr.Blocks() as app:
 
 
     # ✅ Button Actions
-    generate_button.click(generate_script, inputs=[topic_input, duration_input], outputs=[eng_output])
+    # generate_button.click(generate_script, inputs=[topic_input, duration_input], outputs=[eng_output])
+    generate_button.click(generate_script, inputs=[topic_input, duration_input],  outputs=[eng_output])
     extract_keywords_btn.click(extract_keywords, inputs=[eng_output], outputs=[keyword_output])
     translate_button.click(translate_to_urdu, inputs=[eng_output], outputs=[urdu_output])
 
@@ -204,5 +201,12 @@ with gr.Blocks() as app:
     finalize_button = gr.Button("Finalize Process")
     finalize_button.click(finalize_process, outputs=[status_output])
 
+    generate_button.click(
+    lambda topic, duration: (*clear_old_files(), generate_script(topic, duration)),  
+    inputs=[topic_input, duration_input],  
+    outputs=[keyword_output, urdu_output, final_edited_urdu_output, status_output]    )
+
+    
+
 
 app.launch()