rbughao commited on
Commit
e065ed6
·
1 Parent(s): b5aade8

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +28 -0
app.py ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #for learning
2
+ import os
3
+ import openai
4
+ import gradio as gr
5
+
6
+ openai.api_key = os.environ.get('O_APIKey')
7
+
8
+ #using a folder
9
+ from llama_index import VectorStoreIndex, SimpleDirectoryReader, SummaryIndex, download_loader
10
+ PDFReader = download_loader("PDFReader")
11
+ loader = PDFReader()
12
+
13
+ #loading options
14
+ # loader = SimpleDirectoryReader(input_files = ["pdf1","pdf2"])
15
+ # documents = loader.load_data()
16
+ # or
17
+ # documents = loader.load_data(file=['toolkit.pdf','pdf2','pdf3'])
18
+
19
+ documents = loader.load_data(file='cms.pdf')
20
+ index = VectorStoreIndex.from_documents(documents)
21
+ query_engine = index.as_query_engine()
22
+
23
+ def reply(message, history):
24
+ answer = str(query_engine.query(message))
25
+ return answer
26
+
27
+ Conversing = gr.ChatInterface(reply, chatbot=gr.Chatbot(height=500), retry_btn=None,theme=gr.themes.Monochrome(),
28
+ title = 'E-Commerce BT/AN/CA/DH CMS Q&A', undo_btn = None, clear_btn = None, css='footer {visibility: hidden}').launch()