Update app.py

app.py

@@ -1,85 +1,58 @@
-# app.py – Cartoon Explainer Generator (Multilingual)
+# app.py - Subject Explainer Demo (Mock Output with Streamlit)
 
-import os
-import uuid
 import streamlit as st
+import random
+import os
+from moviepy.editor import concatenate_videoclips, TextClip, CompositeVideoClip, AudioFileClip
 from gtts import gTTS
-from transformers import pipeline
-from diffusers import StableDiffusionPipeline
-from PIL import Image
-from moviepy.editor import ImageClip, AudioFileClip, concatenate_videoclips, CompositeAudioClip
-
-# Setup
-st.set_page_config(page_title="🎬 Cartoon Explainer Video Generator", layout="centered")
-st.title("🎨 Cartoon Explainer Video Generator")
-
-# Caching Model Load
-@st.cache_resource(show_spinner=False)
-def load_text_gen():
-    return pipeline("text-generation", model="tiiuae/falcon-rw-1b")
 
-@st.cache_resource(show_spinner=False)
-def load_image_gen():
-    return StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4").to("cpu")
-
-text_gen = load_text_gen()
-image_gen = load_image_gen()
-
-# UI Inputs
+st.set_page_config(page_title="AI Subject Explainer", layout="centered")
+st.title("📘 Realistic Subject Explainer (Demo)")
+
+# Mock story generation
+def generate_detailed_story(topic):
+    samples = {
+        "Newton's Third Law": "In a school cafeteria, two friends are playing air hockey. Every time one hits the puck, the puck pushes back with equal force, gliding smoothly. This shows Newton's Third Law — for every action, there's an equal and opposite reaction.",
+        "Photosynthesis": "Imagine a tiny chef inside a plant leaf, cooking food using sunlight, water, and air. That's photosynthesis! The plant takes sunlight and carbon dioxide to make its food and gives us oxygen in return."
+    }
+    return samples.get(topic, f"Sorry, no story found for '{topic}'. Try 'Newton's Third Law' or 'Photosynthesis'.")
+
+# Create dummy video scene as text clips
+def create_mock_video_scenes(story):
+    scenes = story.split(". ")
+    clips = []
+    for i, scene in enumerate(scenes):
+        txt_clip = TextClip(scene, fontsize=32, color='black', size=(800, 400), bg_color='white', method='caption')
+        txt_clip = txt_clip.set_duration(3)
+        clips.append(txt_clip)
+    return concatenate_videoclips(clips)
+
+# Generate mock voice
+def create_mock_voiceover(story):
+    tts = gTTS(story)
+    audio_path = "mock_voice.mp3"
+    tts.save(audio_path)
+    return audio_path
+
+# Main UI
 topic = st.text_input("Enter a topic (e.g., Newton's Third Law)")
-lang = st.selectbox("Select Narration Language", ["English", "Telugu", "Hindi"])
-generate = st.button("🎥 Generate Cartoon Explainer")
-
-# Language Code Mapping
-lang_map = {"English": "en", "Telugu": "te", "Hindi": "hi"}
-
-if generate and topic:
-    with st.spinner("🧠 Generating story..."):
-        prompt = f"Explain the concept '{topic}' using a creative cartoon-style story for school kids."
-        story_text = text_gen(prompt, max_new_tokens=200, temperature=0.7)[0]["generated_text"]
-        sentences = story_text.strip().split(".")
-        scenes = [s.strip() for s in sentences if len(s.strip()) > 20]
-
-    st.subheader("📖 Story Summary")
-    st.write(story_text)
-
-    audio_path = f"audio_{uuid.uuid4().hex}.mp3"
-    gTTS(text=story_text, lang=lang_map[lang]).save(audio_path)
-
-    with st.spinner("🎨 Generating cartoon scenes..."):
-        image_paths = []
-        for i, scene in enumerate(scenes):
-            img_prompt = f"Cartoon-style illustration of: {scene}"
-            image = image_gen(img_prompt).images[0]
-            img_path = f"scene_{i}_{uuid.uuid4().hex}.png"
-            image.save(img_path)
-            image_paths.append(img_path)
-
-    with st.spinner("🎬 Creating video..."):
-        clips = [
-            ImageClip(path).set_duration(3).fadein(0.3).fadeout(0.3)
-            for path in image_paths
-        ]
-        final_video = concatenate_videoclips(clips, method="compose")
-
-        # Narration
-        narration = AudioFileClip(audio_path)
 
-        # Optional background music
-        music_path = "bg_music.mp3"
-        if os.path.exists(music_path):
-            music = AudioFileClip(music_path).volumex(0.1)
-            background = music.subclip(0, final_video.duration)
-            final_audio = CompositeAudioClip([narration, background])
-        else:
-            final_audio = narration
+if st.button("Generate Video") and topic:
+    with st.spinner("Generating explanation..."):
+        story = generate_detailed_story(topic)
+        st.subheader("🧠 Explanation")
+        st.write(story)
 
-        final_video = final_video.set_audio(final_audio)
+        video = create_mock_video_scenes(story)
+        audio_path = create_mock_voiceover(story)
+        audio = AudioFileClip(audio_path).set_duration(video.duration)
 
-        output_path = f"explainer_{uuid.uuid4().hex}.mp4"
-        final_video.write_videofile(output_path, fps=24, codec="libx264", threads=2, preset="ultrafast")
+        final = CompositeVideoClip([video.set_audio(audio)])
+        output_path = f"final_demo_{random.randint(1000,9999)}.mp4"
+        final.write_videofile(output_path, fps=24, codec='libx264', audio_codec='aac')
 
-    st.subheader("✅ Cartoon Explainer Video")
     st.video(output_path)
-    st.download_button("⬇️ Download Video", open(output_path, "rb").read(), file_name="cartoon_explainer.mp4")
-    st.download_button("🔊 Download Narration", open(audio_path, "rb").read(), file_name="narration.mp3")
+    with open(audio_path, "rb") as a:
+        st.download_button("🔊 Download Audio", a, file_name="voice.mp3")
+    with open(output_path, "rb") as v:
+        st.download_button("🎬 Download Video", v, file_name="explainer_demo.mp4")