Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,100 @@
-import altair as alt
-import numpy as np
-import pandas as pd
 import streamlit as st
+import requests
+import json
+import os
+from dotenv import load_dotenv
+from langchain_community.document_loaders import WebBaseLoader
 
-"""
-# Welcome to Streamlit!
+# Load environment variables from .env file
+load_dotenv()
+
+# 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
+    except Exception as e:
+        st.error(f"Error loading URL: {e}")
+        return None
+
+# Function to summarize text using Gemma 3 27B via OpenRouter API
+def summarize_text(url):
+    """Summarize the content from the given URL using Gemma 3 27B via OpenRouter API"""
+    text = load_text(url)
+    if not text:
+        return None
 
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
+    # Define the prompt for summarization
+    summary_prompt = f"""
+You are an expert summarizer. Your task is to create a concise summary of the following text. The summary should be no more than 7-8 sentences long.
 
-In the meantime, below is an example of what you can do with just a few lines of code:
+TEXT: {text}
+
+SUMMARY:
 """
 
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
-
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
-
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+    try:
+        # Make API request to OpenRouter for summarization
+        response = requests.post(
+            url="https://openrouter.ai/api/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
+            },
+            data=json.dumps({
+                "model": "google/gemma-3-27b-it:free",
+                "messages": [
+                    {
+                        "role": "user",
+                        "content": [
+                            {
+                                "type": "text",
+                                "text": summary_prompt
+                            }
+                        ]
+                    }
+                ]
+            })
+        )
+
+        # Check if the request was successful
+        if response.status_code == 200:
+            result = response.json()
+            summary = result['choices'][0]['message']['content']
+            return summary.strip()
+        else:
+            st.error(f"API Error: {response.status_code} - {response.text}")
+            return None
+
+    except Exception as e:
+        st.error(f"Error summarizing content: {e}")
+        return None
+
+# Streamlit app interface
+st.title("Summarizer AI")
+st.markdown("Enter a URL to summarize the content concisely")
+
+with st.form(key='summarizer_form'):
+    url = st.text_area(
+        label="Enter the URL of the article or blog post:",
+        max_chars=250,
+        placeholder="https://example.com/article"
+    )
+    submit_button = st.form_submit_button(label="Summarize")
+
+if submit_button and url:
+    with st.spinner("Summarizing..."):
+        summary = summarize_text(url)
+        if summary:
+            st.subheader("Summary")
+            st.write(summary)
+        else:
+            st.error("Unable to generate summary. Please check the URL or try again.")