Update app.py

app.py

@@ -1,18 +1,20 @@
+import os
 import nest_asyncio
 nest_asyncio.apply()
 
 import streamlit as st
-import asyncio
-import subprocess
 from transformers import pipeline
+from huggingface_hub import login
 from streamlit.components.v1 import html
 
-# -----------------------------------------------------------
-# Hugging Face CLI Login
-# This will prompt you in the terminal for your Hugging Face token.
-# Ensure you have your token ready.
-subprocess.run(["huggingface-cli", "login"], shell=True)
-# -----------------------------------------------------------
+# Retrieve the token from environment variables
+hf_token = os.environ.get("HF_TOKEN")
+if not hf_token:
+    st.error("Hugging Face token not found. Please set the HF_TOKEN environment variable.")
+    st.stop()
+
+# Login with the token
+login(token=hf_token)
 
 # Initialize session state for timer and results
 if 'result' not in st.session_state:
@@ -30,10 +32,8 @@ def timer():
     (function() {
         var start = Date.now();
         var timerElement = document.getElementById('timer');
-        // Remove any existing freeze flag
         localStorage.removeItem("freezeTimer");
         var interval = setInterval(function() {
-            // If freeze flag is set, stop the timer
             if(localStorage.getItem("freezeTimer") === "true"){
                 clearInterval(interval);
                 timerElement.style.color = '#00cc00';
@@ -57,7 +57,8 @@ st.header("Sentiment Analysis & Report Generation with Gemma")
 @st.cache_resource
 def load_models():
     sentiment_pipe = pipeline("text-classification", model="mixedbread-ai/mxbai-rerank-base-v1")
-    gemma_pipe = pipeline("text-generation", model="google/gemma-3-1b-it")
+    # Pass the token to the Gemma pipeline
+    gemma_pipe = pipeline("text-generation", model="google/gemma-3-1b-it", use_auth_token=hf_token)
     return sentiment_pipe, gemma_pipe
 
 sentiment_pipe, gemma_pipe = load_models()
@@ -69,27 +70,22 @@ if st.button("Generate Report"):
     if not user_input.strip():
         st.error("Please enter some text!")
     else:
-        # Start the timer if not already started
         if not st.session_state.timer_started and not st.session_state.timer_frozen:
             st.session_state.timer_started = True
             html(timer(), height=50)
         
-        # Initialize status message and progress bar
         status_text = st.empty()
         progress_bar = st.progress(0)
         
         try:
-            # ---------------------------
             # Stage 1: Sentiment Analysis
             status_text.markdown("**🔍 Running sentiment analysis...**")
             progress_bar.progress(0)
             sentiment_result = sentiment_pipe(user_input)
             progress_bar.progress(50)
             
-            # ---------------------------
             # Stage 2: Generate Report using Gemma
             status_text.markdown("**📝 Generating report with Gemma...**")
-            # Build a prompt that includes the original text and the sentiment analysis result.
             prompt = f"""
 Generate a detailed report based on the following analysis.
 
@@ -104,17 +100,13 @@ Please provide a concise summary report explaining the sentiment and key insight
             report = gemma_pipe(prompt, max_length=200)
             progress_bar.progress(100)
             status_text.success("**✅ Generation complete!**")
-            
-            # Freeze the timer immediately upon completion.
             html("<script>localStorage.setItem('freezeTimer', 'true');</script>", height=0)
             st.session_state.timer_frozen = True
             
-            # Display the results
             st.write("**Sentiment Analysis Result:**", sentiment_result)
             st.write("**Generated Report:**", report[0]['generated_text'])
             
         except Exception as e:
-            # Remove timer element if an error occurs
             html("<script>document.getElementById('timer').remove();</script>")
             status_text.error(f"**❌ Error:** {str(e)}")
             progress_bar.empty()