Create app.py

app.py

@@ -0,0 +1,63 @@
+import streamlit as st
+from transformers import pipeline
+from diffusers import StableDiffusionPipeline
+import torch
+from PIL import Image
+
+# Title
+st.set_page_config(page_title="AI Meme Generator", page_icon="🎭")
+st.title("🎭 AI Meme Generator (Voice + Text)")
+
+# Load Whisper (speech-to-text)
+@st.cache_resource
+def load_asr():
+    return pipeline("automatic-speech-recognition", model="openai/whisper-tiny")
+
+# Load Stable Diffusion (text-to-image)
+@st.cache_resource
+def load_sd():
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    return StableDiffusionPipeline.from_pretrained(
+        "runwayml/stable-diffusion-v1-5",
+        torch_dtype=torch.float16 if device == "cuda" else torch.float32
+    ).to(device)
+
+asr = load_asr()
+sd_pipe = load_sd()
+
+# Function: Voice -> Text
+def transcribe(audio_file):
+    result = asr(audio_file)
+    return result["text"]
+
+# Function: Text -> Meme
+def generate_meme(prompt):
+    image = sd_pipe(prompt).images[0]
+    return image
+
+# Tabs for input type
+tab1, tab2 = st.tabs(["📝 Text to Meme", "🎤 Voice to Meme"])
+
+# ---------------- Text to Meme ----------------
+with tab1:
+    text_input = st.text_area("Enter your meme idea")
+    if st.button("Generate Meme", key="text_meme"):
+        if text_input.strip():
+            with st.spinner("Generating meme..."):
+                img = generate_meme(f"Meme style funny cartoon with text: {text_input}")
+                st.image(img, caption="Generated Meme")
+        else:
+            st.warning("Please enter some text!")
+
+# ---------------- Voice to Meme ----------------
+with tab2:
+    audio_file = st.file_uploader("Upload your voice recording (mp3/wav)", type=["mp3", "wav"])
+    if st.button("Generate Meme from Voice", key="voice_meme"):
+        if audio_file is not None:
+            with st.spinner("Transcribing and generating meme..."):
+                text = transcribe(audio_file)
+                img = generate_meme(f"Meme style funny cartoon with text: {text}")
+                st.image(img, caption="Generated Meme")
+                st.success(f"Recognized Text: {text}")
+        else:
+            st.warning("Please upload a voice file!")