import gradio as gr
import os
import torch
import yake
import shutil
import glob
import ffmpeg
import cv2
import numpy as np
from datetime import datetime
from gtts import gTTS
from diffusers import StableDiffusionPipeline
from deep_translator import GoogleTranslator
import wikipediaapi
from groq import Groq

# Import functions from your Python files
from script_generator import script_gen_and_translate
from urdu_tts_video import urdu_tts_and_video

def main():
    with gr.Blocks() as app:
        gr.Markdown("# AI-Powered Urdu Science Content Generator")
        
        with gr.Tabs():
            with gr.Tab("Script Generation & Translation"):
                script_gen_and_translate()
            
            with gr.Tab("Urdu TTS & Video Generation"):
                urdu_tts_and_video()
    
    app.launch()

if __name__ == "__main__":
    main()