Spaces:
Sleeping
Sleeping
File size: 2,980 Bytes
415c6f1 68b678c 415c6f1 68b678c 415c6f1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 |
import os
import streamlit as st
import requests
import json
from bs4 import BeautifulSoup
import re
from difflib import SequenceMatcher
# Fix Streamlit permission issue
os.environ['STREAMLIT_HOME'] = '/tmp/.streamlit'
API_URL = "https://jishnusetia-ollama-server.hf.space/api/generate"
MODEL = "llama3.2"
st.title("🌐 Chat with a Website (Ollama)")
# Store website content in memory
if "chunks" not in st.session_state:
st.session_state.chunks = []
if "chat_history" not in st.session_state:
st.session_state.chat_history = []
# Helper: clean and chunk text
def clean_and_chunk(text, chunk_size=800):
text = re.sub(r'\s+', ' ', text).strip()
return [text[i:i+chunk_size] for i in range(0, len(text), chunk_size)]
# Helper: get top N relevant chunks
def get_relevant_chunks(query, chunks, top_n=3):
scored = [(SequenceMatcher(None, query, chunk).ratio(), chunk) for chunk in chunks]
scored.sort(key=lambda x: x[0], reverse=True)
return [chunk for _, chunk in scored[:top_n]]
# Step 1: User inputs website
url = st.text_input("Enter website URL:")
if st.button("Fetch Website") and url:
try:
resp = requests.get(url, timeout=10)
soup = BeautifulSoup(resp.text, 'html.parser')
for script in soup(["script", "style"]):
script.extract()
text = soup.get_text()
st.session_state.chunks = clean_and_chunk(text)
st.success(f"Website fetched! {len(st.session_state.chunks)} chunks ready.")
except Exception as e:
st.error(f"Error fetching website: {e}")
# Step 2: Function to send query to Ollama
def send_message(message, context):
prompt = f"Answer the following question based on this context:\n{context}\n\nQuestion: {message}"
payload = {"model": MODEL, "prompt": prompt}
response = requests.post(API_URL, json=payload, stream=True)
if response.ok:
reply = ""
for line in response.iter_lines():
if line:
data = json.loads(line.decode('utf-8'))
reply += data.get("response", "")
return reply.strip()
else:
return "⚠️ Error: Failed to get response from Ollama API."
# Step 3: Chat input
user_input = st.text_input("Ask a question about the website:")
if st.button("Send Question") and user_input.strip():
if not st.session_state.chunks:
st.error("Please fetch a website first.")
else:
st.session_state.chat_history.append(("You", user_input))
with st.spinner("Thinking..."):
context = "\n".join(get_relevant_chunks(user_input, st.session_state.chunks))
bot_reply = send_message(user_input, context)
st.session_state.chat_history.append(("Ollama", bot_reply))
# Step 4: Display chat history
st.subheader("Chat History")
for speaker, message in st.session_state.chat_history:
if speaker == "You":
st.markdown(f"**You:** {message}")
else:
st.markdown(f"**🤖 Ollama:** {message}")
|