datascientist22 commited on
Commit
7d6baca
·
verified ·
1 Parent(s): 0f09491

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -7
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 and url_input:
 
 
 
 
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)