Create app.py

app.py

@@ -0,0 +1,103 @@
+import streamlit as st
+from groq import Groq
+import os
+from dotenv import load_dotenv
+from langchain_groq import ChatGroq
+from langchain import hub
+from langchain_chroma import Chroma
+from langchain_community.document_loaders import WebBaseLoader
+from langchain_core.output_parsers import StrOutputParser
+from langchain_core.runnables import RunnablePassthrough
+from langchain_text_splitters import RecursiveCharacterTextSplitter
+from sentence_transformers import SentenceTransformer
+import bs4
+import torch
+
+# Initialize the Groq client
+client = Groq(api_key="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")
+
+def generate_response_groq(context, query):
+    """Generate response using Groq API."""
+    prompt = f"Context: {context}\nQuestion: {query}\nAnswer:"
+    chat_completion = client.chat.completions.create(
+        messages=[
+            {
+                "role": "user",
+                "content": prompt,
+            }
+        ],
+        model="llama3-8b-8192",
+    )
+    response = chat_completion.choices[0].message.content
+    return response
+
+def extract_text_from_url(url):
+    """Extract text from URL using WebBaseLoader."""
+    loader = WebBaseLoader(
+        web_paths=(url,),
+        bs_kwargs=dict(
+            parse_only=bs4.SoupStrainer()
+        ),
+    )
+    docs = loader.load()
+    text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
+    splits = text_splitter.split_documents(docs)
+    text = ""
+    for split in splits:
+        text += split.page_content
+    return text
+
+# Set the page layout to wide for better UI space
+st.set_page_config(page_title="URL Query Application", layout="wide")
+
+# # Sidebar
+# st.sidebar.title("URL Query Assistant")
+# st.sidebar.image("https://upload.wikimedia.org/wikipedia/commons/thumb/8/87/Globe_icon_2.svg/2048px-Globe_icon_2.svg.png", use_column_width=True)  # Adding an image in the sidebar
+# st.sidebar.markdown("### Navigation")
+# st.sidebar.markdown("Use this app to ask questions about the content of a predefined URL.")
+# st.sidebar.markdown("")
+
+# Main UI layout
+st.title("🌐 URL Query Application")
+st.markdown("""
+<style>
+    .main-content {background-color: #f0f2f6; padding: 20px; border-radius: 10px;}
+    .stButton>button {background-color: #4CAF50; color: white; font-size: 16px; border-radius: 10px;}
+    .stTextInput>div>div>input {background-color: #f0f2f6; color: black; border-radius: 5px;}
+</style>
+""", unsafe_allow_html=True)
+
+st.markdown("<div class='main-content'>", unsafe_allow_html=True)
+
+url = "https://iesco.com.pk/index.php/customer-services/tariff-guide#"
+document_text = extract_text_from_url(url)
+st.text_area("📜 Extracted Text", document_text, height=200)
+
+query = st.text_input("🔍 Enter your query")
+
+if st.button("💬 Get Answer"):
+    if query:
+        with st.spinner("Generating response..."):
+            response = generate_response_groq(document_text, query)
+            st.write("**Response:**")
+            st.write(response)
+    else:
+        st.error("Please enter a query.")
+
+st.markdown("</div>", unsafe_allow_html=True)
+
+# # Footer
+# st.sidebar.markdown("### About")
+# st.sidebar.info("Developed using Streamlit and Groq API.")
+# st.sidebar.markdown("---")
+# st.sidebar.write("For more information, visit [Groq](https://www.groq.com) and [Streamlit](https://streamlit.io).")
+
+# Customize the theme and color contrast
+st.markdown("""
+<style>
+    .css-1aumxhk {background-color: #E8EAF6;}
+    .stTextInput>div>div>input {border-color: #3f51b5;}
+    .stTextArea>div>div>textarea {border-color: #3f51b5;}
+    .stButton>button {background-color: #3f51b5; color: white; font-size: 16px;}
+</style>
+""", unsafe_allow_html=True)