Update pages/model.py

pages/model.py

@@ -4,54 +4,75 @@ from langchain.prompts import ChatPromptTemplate, SystemMessagePromptTemplate, H
 from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
 from langchain_community.document_loaders import YoutubeLoader
 
-HF_Token = os.getenv("HUGGINGFACEHUB_API_TOKEN")  
+hf_token = os.getenv("HF_Token")
+if hf_token is None:
+    st.error("🔐 Please set the 'HF_Token' environment variable with your Hugging Face token.")
+else:
+    os.environ["HUGGINGFACEHUB_API_KEY"] = hf_token
 
-st.set_page_config(page_title="Tube2Insight 🎬", layout="centered")
-st.markdown("## 🎥 Tube2Insight\n*Summarize any YouTube video using LLaMA power!*")
+st.set_page_config(page_title="🎬 Tube2Insight", page_icon="🎥", layout="wide")
 
-video_url = st.text_input("📺 Paste a YouTube video link")
+st.markdown("""
+    <style>
+        .stApp { background-color: #f7f9fc; }
+        .video-card {
+            background-color: #ffffff;
+            border: 2px solid #e0e0e0;
+            padding: 20px;
+            border-radius: 15px;
+            text-align: center;
+            box-shadow: 2px 2px 10px #ddd;
+        }
+        .video-card:hover {
+            border-color: #FF4B4B;
+            box-shadow: 3px 3px 12px #bbb;
+        }
+    </style>
+""", unsafe_allow_html=True)
 
-with st.expander("ℹ️ How It Works", expanded=False):
-    st.markdown("""
-    1. Enter the YouTube URL 🎯  
-    2. The app grabs the transcript 📝  
-    3. LLaMA summarizes it using smart prompts 🦙  
-    4. You get a quick summary without watching the whole video! ⚡  
-    """)
+st.markdown("<h1 style='text-align: center; color: #FF4B4B;'>🎬 Tube2Insight</h1>", unsafe_allow_html=True)
+st.markdown("### Paste a YouTube link below to get an instant summary 👇")
 
+video_url = st.text_input("📺 YouTube Video URL")
 
-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
+if st.button("🔍 Summarize"):
+    if not video_url:
+        st.warning("Please provide a video link.")
+    else:
+        with st.spinner("📄 Fetching transcript..."):
+            try:
+                docs = YoutubeLoader.from_youtube_url(video_url).load()
+                transcript = docs[0].page_content
+                st.success("✅ Transcript loaded!")
+            except Exception as e:
+                st.error(f"Error loading video: {e}")
+                st.stop()
 
-        with st.expander("📄 View Full Transcript"):
+        with st.expander("📄 View 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}
+            model_kwargs={"temperature": 0.5, "max_new_tokens": 300}
         )
         
         prompt = ChatPromptTemplate.from_messages([
-            ("system", "You are a helpful assistant that summarizes YouTube transcripts."),
-            ("human", "Summarize the following transcript:\n\n{transcript}")
+            SystemMessagePromptTemplate.from_template("You are a helpful assistant summarizing YouTube transcripts."),
+            HumanMessagePromptTemplate.from_template("{transcript}")
         ])
-
         messages = prompt.format_messages(transcript=transcript)
-        response = llm(messages)
 
-        st.success("✅ Summary Ready!")
-        st.markdown("### 📝 Summary")
-        st.write(response.content)
+        with st.spinner("🧠 Summarizing with LLaMA..."):
+            response = llm.invoke(messages)
+
+        st.markdown("### 📌 Summary")
+        st.success(response.content)
 
-    except Exception as e:
-        st.error(f"❌ Error: {e}")
+st.markdown("---")
+st.markdown(
+    "<p style='text-align: center; color: gray;'>🚀 Built with LangChain + LLaMA + Streamlit | by <a href='https://github.com/yourusername' target='_blank'>YourName</a></p>",
+    unsafe_allow_html=True
+)