Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,15 @@
|
|
1 |
import streamlit as st
|
2 |
import re
|
3 |
import os
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
4 |
|
5 |
# Sidebar Style with Multicolored Background
|
6 |
sidebar_bg_style = """
|
@@ -81,14 +90,13 @@ if 'chat_history' not in st.session_state:
|
|
81 |
|
82 |
# Submit button for chat
|
83 |
if st.button("Submit Query"):
|
84 |
-
if query
|
|
|
|
|
|
|
|
|
85 |
# Blog loading logic based on user input URL
|
86 |
-
loader = WebBaseLoader
|
87 |
-
web_paths=(url_input,), # Use the user-input URL
|
88 |
-
bs_kwargs=dict(
|
89 |
-
parse_only=bs4.SoupStrainer() # Adjust based on the user's URL structure
|
90 |
-
),
|
91 |
-
)
|
92 |
docs = loader.load()
|
93 |
|
94 |
text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
|
|
|
1 |
import streamlit as st
|
2 |
import re
|
3 |
import os
|
4 |
+
import bs4
|
5 |
+
from sentence_transformers import SentenceTransformer
|
6 |
+
import torch
|
7 |
+
from langchain.chains import ChatGroq
|
8 |
+
from langchain.document_loaders import UnstructuredURLLoader # Updated import
|
9 |
+
from langchain.prompts import RunnablePassthrough
|
10 |
+
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
11 |
+
from langchain.vectorstores import Chroma
|
12 |
+
from langchain.output_parsers import StrOutputParser
|
13 |
|
14 |
# Sidebar Style with Multicolored Background
|
15 |
sidebar_bg_style = """
|
|
|
90 |
|
91 |
# Submit button for chat
|
92 |
if st.button("Submit Query"):
|
93 |
+
if not query:
|
94 |
+
st.markdown('<p style="color:red; font-weight:bold;">Please enter a query before submitting</p>', unsafe_allow_html=True)
|
95 |
+
elif not url_input:
|
96 |
+
st.markdown('<p style="color:red; font-weight:bold;">Please enter a valid URL before submitting</p>', unsafe_allow_html=True)
|
97 |
+
else:
|
98 |
# Blog loading logic based on user input URL
|
99 |
+
loader = UnstructuredURLLoader(urls=[url_input]) # Using UnstructuredURLLoader instead of WebBaseLoader
|
|
|
|
|
|
|
|
|
|
|
100 |
docs = loader.load()
|
101 |
|
102 |
text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
|