shallou commited on
Commit
12fc98c
·
verified ·
1 Parent(s): b1b9fc9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -8
app.py CHANGED
@@ -1,11 +1,11 @@
1
- import os
 
2
  import pickle
3
- import numpy as np
4
  from PyPDF2 import PdfReader
5
  from transformers import pipeline
6
  from sentence_transformers import SentenceTransformer
7
- from dotenv import load_dotenv
8
- import streamlit as st
9
 
10
  # Load environment variables from .env file
11
  load_dotenv()
@@ -22,6 +22,7 @@ def chunk_text(text, chunk_size=1000, chunk_overlap=200):
22
  # Function to generate embeddings using sentence-transformers
23
  def generate_embeddings(text_chunks, model_name='all-MiniLM-L6-v2'):
24
  model = SentenceTransformer(model_name)
 
25
  embeddings = model.encode(text_chunks, convert_to_tensor=False)
26
  return embeddings
27
 
@@ -42,6 +43,7 @@ def main():
42
 
43
  if pdf is not None:
44
  pdf_reader = PdfReader(pdf)
 
45
  text = ""
46
  for page in pdf_reader.pages:
47
  text += page.extract_text()
@@ -78,8 +80,8 @@ def main():
78
  result = qa_pipeline(question=query, context=best_chunk)
79
  st.write(result['answer'])
80
 
81
- # Set background image from URL
82
- set_bg_from_url("https://www.1access.com/wp-content/uploads/2019/10/GettyImages-1180389186.jpg", opacity=0.5)
83
 
84
  def set_bg_from_url(url, opacity=1):
85
  footer = """
@@ -105,5 +107,20 @@ def set_bg_from_url(url, opacity=1):
105
  </footer>
106
  """
107
  st.markdown(footer, unsafe_allow_html=True)
108
-
109
- # Set background image using
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from dotenv import load_dotenv
2
+ import streamlit as st
3
  import pickle
 
4
  from PyPDF2 import PdfReader
5
  from transformers import pipeline
6
  from sentence_transformers import SentenceTransformer
7
+ import os
8
+ import numpy as np
9
 
10
  # Load environment variables from .env file
11
  load_dotenv()
 
22
  # Function to generate embeddings using sentence-transformers
23
  def generate_embeddings(text_chunks, model_name='all-MiniLM-L6-v2'):
24
  model = SentenceTransformer(model_name)
25
+
26
  embeddings = model.encode(text_chunks, convert_to_tensor=False)
27
  return embeddings
28
 
 
43
 
44
  if pdf is not None:
45
  pdf_reader = PdfReader(pdf)
46
+
47
  text = ""
48
  for page in pdf_reader.pages:
49
  text += page.extract_text()
 
80
  result = qa_pipeline(question=query, context=best_chunk)
81
  st.write(result['answer'])
82
 
83
+ if __name__ == '__main__':
84
+ main()
85
 
86
  def set_bg_from_url(url, opacity=1):
87
  footer = """
 
107
  </footer>
108
  """
109
  st.markdown(footer, unsafe_allow_html=True)
110
+
111
+ # Set background image using HTML and CSS
112
+ st.markdown(
113
+ f"""
114
+ <style>
115
+ body {{
116
+ background: url('{url}') no-repeat center center fixed;
117
+ background-size: cover;
118
+ opacity: {opacity};
119
+ }}
120
+ </style>
121
+ """,
122
+ unsafe_allow_html=True
123
+ )
124
+
125
+ # Set background image from URL
126
+ set_bg_from_url("https://www.1access.com/wp-content/uploads/2019/10/GettyImages-1180389186.jpg", opacity=0.