Update pages/model.py

pages/model.py

@@ -2,12 +2,12 @@ import os
 import streamlit as st
 from langchain.prompts import ChatPromptTemplate, SystemMessagePromptTemplate, HumanMessagePromptTemplate
 from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
+from langchain_community.document_loaders import YoutubeLoader
 
-os.environ["HUGGINGFACEHUB_API_KEY"] = os.getenv('HF_Token')
-os.environ["HF_TOKEN"] = os.getenv('HF_Token')
+HF_Token = os.getenv("HUGGINGFACEHUB_API_TOKEN")  
 
 st.set_page_config(page_title="Tube2Insight 🎬", layout="centered")
-st.markdown("## 🎥 Tube2Insight\n*Summarize any YouTube video using LLM!*")
+st.markdown("## 🎥 Tube2Insight\n*Summarize any YouTube video using LLaMA power!*")
 
 video_url = st.text_input("📺 Paste a YouTube video link")
 
@@ -19,5 +19,41 @@ with st.expander("ℹ️ How It Works", expanded=False):
     4. You get a quick summary without watching the whole video! ⚡  
     """)
 
+if not hf_token:
+    st.warning("⚠️ Please set `HUGGINGFACEHUB_API_TOKEN` in your environment.", icon="🔐")
+
+if st.button("🚀 Generate Summary") and video_url and hf_token:
+    try:
+        st.info("⏳ Fetching transcript from YouTube...")
+        loader = YoutubeLoader.from_youtube_url(video_url)
+        docs = loader.load()
+        transcript = docs[0].page_content
+
+        with st.expander("📄 View Full Transcript"):
+            st.write(transcript)
+
+        st.info("🧠 Running LLaMA summarization...")
+
+        llm = ChatHuggingFace(
+            repo_id="meta-llama/Llama-3.1-8B-Instruct",
+            huggingfacehub_api_token=HF_Token,
+            model_kwargs={"temperature": 0.5, "max_new_tokens": 512}
+        )
+        
+        prompt = ChatPromptTemplate.from_messages([
+            ("system", "You are a helpful assistant that summarizes YouTube transcripts."),
+            ("human", "Summarize the following transcript:\n\n{transcript}")
+        ])
+
+        messages = prompt.format_messages(transcript=transcript)
+        response = llm(messages)
+
+        st.success("✅ Summary Ready!")
+        st.markdown("### 📝 Summary")
+        st.write(response.content)
+
+    except Exception as e:
+        st.error(f"❌ Error: {e}")
+