Create app.py

app.py

@@ -0,0 +1,174 @@
+import os
+import datetime
+import gradio as gr
+from groq import Groq
+import torch
+from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
+
+
+# ✅ Set API Key
+os.environ["GROQ_API_KEY"] = "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"  # Replace with your API key
+# GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
+
+client = Groq(api_key=os.environ["GROQ_API_KEY"])
+
+
+# ✅ Load M2M-100 Model
+model_name = "facebook/m2m100_418M"
+tokenizer = M2M100Tokenizer.from_pretrained(model_name)
+model = M2M100ForConditionalGeneration.from_pretrained(model_name)
+
+# ✅ Function to Get Response from Groq API
+def get_groq_response(prompt):
+    print("🤖 Generating Response...")
+    client = Groq(api_key=os.environ["GROQ_API_KEY"])
+    chat_completion = client.chat.completions.create(
+        messages=[{"role": "user", "content": prompt}],
+        model="llama-3.3-70b-versatile"
+    )
+    return chat_completion.choices[0].message.content
+
+def generate_script(topic, duration):
+    try:
+        words_per_minute = 130  # Approximate speaking rate
+        target_words = duration * words_per_minute
+
+        response = client.chat.completions.create(
+            messages=[
+                {"role": "user", "content": f"Generate a plain, well-structured educational script about {topic} in English with approximately {target_words} words. Do not include stage directions, sound cues, or visual elements—only the spoken content."}
+            ],
+            model="llama-3.3-70b-versatile"
+        )
+
+
+        english_script = response.choices[0].message.content
+        return english_script
+    except Exception as e:
+      print(f"❌ Error in script generation: {str(e)}")
+      return ""
+
+
+def translate_to_urdu(english_script):
+    try:
+        tokenizer.src_lang = "en"
+        inputs = tokenizer(english_script, return_tensors="pt", truncation=True, max_length=1024).to(model.device)
+
+        translated_tokens = model.generate(
+            **inputs,
+            max_length=1024,
+            no_repeat_ngram_size=2,  # Reduce repetition checks (faster)
+            forced_bos_token_id=tokenizer.get_lang_id("ur"),
+            num_beams=2  # Faster than beam search with high values
+        )
+
+        urdu_script = tokenizer.decode(translated_tokens[0], skip_special_tokens=True)
+        return urdu_script
+
+    except Exception as e:
+        return f"❌ Error in translation: {str(e)}"
+
+# # ✅ Function to Translate English Script to Urdu
+# def translate_to_urdu(english_script):
+#     try:
+#         tokenizer.src_lang = "en"
+#         inputs = tokenizer(english_script, return_tensors="pt", truncation=True, max_length=1024).to(model.device)
+
+#         translated_tokens = model.generate(
+#             **inputs,
+#             max_length=1024,
+#             no_repeat_ngram_size=2,  # Reduce repetition checks (faster)
+#             forced_bos_token_id=tokenizer.lang_code_to_id["ur"],  # Fix for language ID error
+#             num_beams=2  # Faster translation
+#         )
+
+#         urdu_script = tokenizer.decode(translated_tokens[0], skip_special_tokens=True)
+#         return urdu_script
+#     except Exception as e:
+#         return f"❌ Error in translation: {str(e)}"
+
+
+
+# ✅ Function to Save Edited Urdu Script to a File
+def save_edited_urdu(edited_urdu_script, topic):
+    timestamp = datetime.datetime.now().strftime("%Y_%m_%d")
+    filename = f"{topic}_Urdu_Final_{timestamp}.txt"
+
+    with open(filename, "w", encoding="utf-8") as file:
+        file.write(edited_urdu_script)
+
+    return filename  # Return the file name so it can be downloaded
+
+
+
+
+def save_file(content, filename):
+    with open(filename, "w", encoding="utf-8") as file:
+        file.write(content)
+    return filename
+
+def process_request(topic, duration):
+    eng_script = generate_script(topic, duration)
+    return eng_script, ""
+
+def process_translation(eng_script):
+    urdu_script = translate_to_urdu(eng_script)
+    return urdu_script
+
+def download_english(eng_script, topic):
+    timestamp = datetime.datetime.now().strftime("%Y_%m_%d")
+    filename = f"{topic}_Eng_{timestamp}.txt"
+    return save_file(eng_script, filename)
+
+def download_urdu(urdu_script, topic):
+    timestamp = datetime.datetime.now().strftime("%Y_%m_%d")
+    filename = f"{topic}_Urdu_{timestamp}.txt"
+    return save_file(urdu_script, filename)
+
+def download_final_urdu(edited_urdu_script, topic):
+    timestamp = datetime.datetime.now().strftime("%Y_%m_%d")
+    filename = f"{topic}_Urdu_Final_{timestamp}.txt"
+    return save_file(edited_urdu_script, filename)
+
+# ✅ Gradio UI
+with gr.Blocks() as app:
+    gr.Markdown("# 🎬 AI-Powered Script Generator")
+
+    topic_input = gr.Textbox(label="Enter Topic")
+    duration_input = gr.Slider(minimum=1, maximum=30, step=1, label="Duration (minutes)")
+    generate_button = gr.Button("Generate English Script")
+    eng_output = gr.Textbox(label="Generated English Script", interactive=False)
+    generate_button.click(process_request, inputs=[topic_input, duration_input], outputs=[eng_output])
+
+    translate_button = gr.Button("Generate Urdu Script")
+    urdu_output = gr.Textbox(label="Translated Urdu Script (Editable)", interactive=True)
+    translate_button.click(process_translation, inputs=[eng_output], outputs=[urdu_output])
+
+    # Editable Urdu Textbox
+    gr.Markdown("### ✍ Edit Urdu Script Below")
+    editable_urdu = gr.Textbox(label="Edited Urdu Script", interactive=True)
+    # Save Edited Urdu Script
+    save_button = gr.Button("Save Edited Urdu Script")
+
+    # Update the editable Urdu box with translated script
+    translate_button.click(lambda x: x, inputs=[urdu_output], outputs=[editable_urdu])
+
+
+    download_eng_button = gr.Button("📥 Download English Script")
+    download_eng_file = gr.File()
+
+    download_urdu_button = gr.Button("📥 Download Translated Urdu Script")
+    download_urdu_file = gr.File()
+
+    # Download Edited Urdu Script
+    download_final_urdu_button = gr.Button("📥 Download Edited Urdu Script")
+    download_final_urdu_file = gr.File()
+
+    # Save the edited script
+    save_button.click(save_edited_urdu, inputs=[editable_urdu, topic_input], outputs=[download_final_urdu_file])
+
+
+    download_eng_button.click(download_english, inputs=[eng_output, topic_input], outputs=[download_eng_file])
+    download_urdu_button.click(download_urdu, inputs=[urdu_output, topic_input], outputs=[download_urdu_file])
+    download_final_urdu_button.click(download_final_urdu, inputs=[editable_urdu, topic_input], outputs=[download_final_urdu_file])
+
+app.launch()