NAB1108 commited on
Commit
4eea648
·
1 Parent(s): ec6e745

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -6
app.py CHANGED
@@ -27,8 +27,9 @@ from langchain.llms import OpenAI
27
  from langchain.document_loaders import PyPDFLoader
28
  from langchain.indexes import VectorstoreIndexCreator
29
  import tempfile
 
30
 
31
- os.environ["OPENAI_API_KEY"] = os.environ['OpenApi_Key']
32
  query1=" "
33
  limit = 0
34
  def loading_pdf():
@@ -36,11 +37,14 @@ def loading_pdf():
36
 
37
  def pdf_changes(pdf_doc, prompt):
38
 
 
 
39
  #loader = OnlinePDFLoader(pdf_doc.name)
40
- loader = PyPDFLoader(pdf_doc.name)
41
- documents = loader.load()
 
42
 
43
- name_filter = "**/*.md"
44
  separator = "\n"
45
  chunk_size_limit = 1000
46
  max_chunk_overlap = 50
@@ -95,7 +99,7 @@ def bot(history):
95
  def infer(question):
96
  global query1
97
  global limit
98
- openai.api_key = os.environ['OpenApi_Key']
99
  prompt_text = question
100
  if prompt_text:
101
  query1 = query1 + "\nUser: " + prompt_text + "\nBot: "
@@ -128,7 +132,8 @@ with gr.Blocks(css=css,theme = gr.themes.Soft()) as demo:
128
  #gr.HTML(title)
129
  with gr.Row():
130
  with gr.Column(scale=1):
131
- pdf_doc = gr.File(label="Load a pdf", file_types=['.pdf'], type="file")
 
132
  prompt = gr.Textbox(label="Behaviour Prompt (optional)", placeholder="Reply to all questions as a rap / Reply to all questions in Hindi etc. ")
133
  #repo_id = gr.Dropdown(label="LLM", choices=["google/flan-ul2", "OpenAssistant/oasst-sft-1-pythia-12b", "bigscience/bloomz"], value="google/flan-ul2")
134
  with gr.Row():
 
27
  from langchain.document_loaders import PyPDFLoader
28
  from langchain.indexes import VectorstoreIndexCreator
29
  import tempfile
30
+ from langchain.document_loaders import DirectoryLoader
31
 
32
+ os.environ["OPENAI_API_KEY"] = "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"
33
  query1=" "
34
  limit = 0
35
  def loading_pdf():
 
37
 
38
  def pdf_changes(pdf_doc, prompt):
39
 
40
+ loader1 = DirectoryLoader(pdf_doc[0].name)
41
+ loader2 = DirectoryLoader(pdf_doc[1].name)
42
  #loader = OnlinePDFLoader(pdf_doc.name)
43
+ #loader = PyPDFLoader(pdf_doc.name)
44
+ documents = loader1.load() + loader2.load()
45
+ print(len(documents))
46
 
47
+ #name_filter = "**/*.md"
48
  separator = "\n"
49
  chunk_size_limit = 1000
50
  max_chunk_overlap = 50
 
99
  def infer(question):
100
  global query1
101
  global limit
102
+ openai.api_key = "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"
103
  prompt_text = question
104
  if prompt_text:
105
  query1 = query1 + "\nUser: " + prompt_text + "\nBot: "
 
132
  #gr.HTML(title)
133
  with gr.Row():
134
  with gr.Column(scale=1):
135
+ #gr.File(file_count="multiple")
136
+ pdf_doc = gr.File(label="Load a pdf", file_count="multiple")
137
  prompt = gr.Textbox(label="Behaviour Prompt (optional)", placeholder="Reply to all questions as a rap / Reply to all questions in Hindi etc. ")
138
  #repo_id = gr.Dropdown(label="LLM", choices=["google/flan-ul2", "OpenAssistant/oasst-sft-1-pythia-12b", "bigscience/bloomz"], value="google/flan-ul2")
139
  with gr.Row():