Upload 3 files

src/app.py

@@ -0,0 +1,22 @@
+import streamlit as st
+from utils import generate_script
+
+st.title("🎬 YouTube Script Generator (Hugging Face Version)")
+
+prompt = st.text_input("Enter your video topic:")
+video_length = st.slider("Video length (minutes):", 1, 20, 5)
+creativity = st.slider("Creativity:", 0.0, 1.0, 0.7)
+
+if st.button("Generate Script"):
+    with st.spinner("Generating script..."):
+        try:
+            search_result, title, script = generate_script(prompt, video_length, creativity)
+            st.success("✅ Script Generated Successfully!")
+            st.subheader("🎯 Title:")
+            st.write(title)
+            st.subheader("🔍 Search Data:")
+            st.write(search_result)
+            st.subheader("📝 Script:")
+            st.write(script)
+        except Exception as e:
+            st.error(f"Error: {e}")

src/utils.py

@@ -0,0 +1,48 @@
+from langchain_community.llms import HuggingFaceHub
+from langchain.prompts import PromptTemplate
+from langchain.chains import LLMChain
+from langchain_community.tools import DuckDuckGoSearchRun
+import os
+
+
+def generate_script(prompt, video_length, creativity):
+    """Generates a YouTube video script using a Hugging Face model (no OpenAI)."""
+
+    api_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
+    if not api_token:
+        raise ValueError("❌ Missing HUGGINGFACEHUB_API_TOKEN — please set it in Hugging Face Space Secrets.")
+
+    # ✅ Lightweight, free-tier-compatible model
+    llm = HuggingFaceHub(
+        repo_id="google/flan-t5-base",
+        model_kwargs={
+            "temperature": float(creativity),
+            "max_length": 512
+        }
+    )
+
+    title_template = PromptTemplate(
+        input_variables=["subject"],
+        template="Write a catchy YouTube video title about {subject}."
+    )
+
+    script_template = PromptTemplate(
+        input_variables=["title", "DuckDuckGo_Search", "duration"],
+        template=(
+            "Write a full YouTube video script for the title '{title}'. "
+            "The video should last about {duration} minutes. "
+            "Include relevant details based on this info: {DuckDuckGo_Search}."
+        )
+    )
+
+    title_chain = LLMChain(llm=llm, prompt=title_template)
+    script_chain = LLMChain(llm=llm, prompt=script_template)
+
+    search = DuckDuckGoSearchRun()
+
+    # Generate title and script
+    title = title_chain.run(prompt)
+    search_result = search.run(prompt)
+    script = script_chain.run(title=title, DuckDuckGo_Search=search_result, duration=video_length)
+
+    return search_result, title, script