Update src/streamlit_app.py

src/streamlit_app.py

@@ -5,27 +5,30 @@ import os
 from dotenv import load_dotenv
 from langchain_community.document_loaders import WebBaseLoader
 
-# Load environment variables from .env file
+# Load environment variables
 load_dotenv()
 
+# Debug: Check if API key is loaded
+if not os.getenv('GROQ_API_KEY'):
+    st.error("GROQ_API_KEY not found in environment variables. Please check your .env file.")
+
 # Function to load the article/blog post from a URL
 def load_text(url):
     """Load the article/blog post from a URL"""
     try:
         loader = WebBaseLoader(url)
         loader.requests_kwargs = {
-            'verify': False,
             'headers': {'User-Agent': os.getenv('USER_AGENT', 'SummarizerBot/1.0 (https://your-site.com)')}
         }
         docs = loader.load()
-        return docs[0].page_content if docs else None  # Extract text content
+        return docs[0].page_content if docs else None
     except Exception as e:
         st.error(f"Error loading URL: {e}")
         return None
 
-# Function to summarize text using Gemma 3 27B via OpenRouter API
+# Function to summarize text using Llama 2 70B via Groq API
 def summarize_text(url):
-    """Summarize the content from the given URL using Gemma 3 27B via OpenRouter API"""
+    """Summarize the content from the given URL using Llama 2 70B via Groq API"""
     text = load_text(url)
     if not text:
         return None
@@ -40,28 +43,23 @@ SUMMARY:
 """
 
     try:
-        # Make API request to OpenRouter for summarization
+        # Make API request to Groq for summarization
         response = requests.post(
-            url="https://openrouter.ai/api/v1/chat/completions",
+            url="https://api.groq.com/openai/v1/chat/completions",
             headers={
-                "Authorization": f"Bearer {os.getenv('OPENROUTER_API_KEY')}",
-                "Content-Type": "application/json",
-                "HTTP-Referer": os.getenv('SITE_URL', '<YOUR_SITE_URL>'),  # Optional, defaults to placeholder
-                "X-Title": os.getenv('SITE_NAME', '<YOUR_SITE_NAME>'),  # Optional, defaults to placeholder
+                "Authorization": f"Bearer {os.getenv('GROQ_API_KEY')}",
+                "Content-Type": "application/json"
             },
             data=json.dumps({
-                "model": "google/gemma-3-27b-it:free",
+                "model": "llama2-70b-4096",
                 "messages": [
                     {
                         "role": "user",
-                        "content": [
-                            {
-                                "type": "text",
-                                "text": summary_prompt
-                            }
-                        ]
+                        "content": summary_prompt
                     }
-                ]
+                ],
+                "max_tokens": 500,  # Limit output for concise summaries
+                "temperature": 0.7  # Balanced creativity for summarization
             })
         )