Update app.py
Browse files
app.py
CHANGED
|
@@ -1,14 +1,10 @@
|
|
| 1 |
import os
|
| 2 |
-
from PyPDF2 import PdfReader
|
| 3 |
-
import docx
|
| 4 |
from langchain.text_splitter import CharacterTextSplitter
|
| 5 |
from langchain.embeddings.openai import OpenAIEmbeddings
|
| 6 |
from langchain.vectorstores import FAISS
|
| 7 |
from langchain.chains.question_answering import load_qa_chain
|
| 8 |
from langchain_openai import OpenAI
|
| 9 |
-
from langchain.callbacks import get_openai_callback
|
| 10 |
import gradio as gr
|
| 11 |
-
from aiohttp import web
|
| 12 |
|
| 13 |
# Set your OpenAI API key
|
| 14 |
os.environ["OPENAI_API_KEY"] = "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"
|
|
@@ -42,19 +38,10 @@ llm = OpenAI()
|
|
| 42 |
chain = load_qa_chain(llm, chain_type="stuff")
|
| 43 |
|
| 44 |
# Define the chatbot interface
|
| 45 |
-
|
| 46 |
-
data = await request.post()
|
| 47 |
-
input_text = data.get("input_text", "")
|
| 48 |
-
|
| 49 |
docs = docsearch.similarity_search(input_text)
|
| 50 |
response = chain.run(input_documents=docs, question=input_text)
|
| 51 |
-
|
| 52 |
-
return web.Response(text=response)
|
| 53 |
|
| 54 |
-
|
| 55 |
-
|
| 56 |
-
app.router.add_post('/chatbot', chatbot_interface)
|
| 57 |
-
|
| 58 |
-
# Run the web server
|
| 59 |
-
if __name__ == "__main__":
|
| 60 |
-
web.run_app(app, port=os.getenv("PORT", 8080))
|
|
|
|
| 1 |
import os
|
|
|
|
|
|
|
| 2 |
from langchain.text_splitter import CharacterTextSplitter
|
| 3 |
from langchain.embeddings.openai import OpenAIEmbeddings
|
| 4 |
from langchain.vectorstores import FAISS
|
| 5 |
from langchain.chains.question_answering import load_qa_chain
|
| 6 |
from langchain_openai import OpenAI
|
|
|
|
| 7 |
import gradio as gr
|
|
|
|
| 8 |
|
| 9 |
# Set your OpenAI API key
|
| 10 |
os.environ["OPENAI_API_KEY"] = "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"
|
|
|
|
| 38 |
chain = load_qa_chain(llm, chain_type="stuff")
|
| 39 |
|
| 40 |
# Define the chatbot interface
|
| 41 |
+
def chatbot_interface(input_text):
|
|
|
|
|
|
|
|
|
|
| 42 |
docs = docsearch.similarity_search(input_text)
|
| 43 |
response = chain.run(input_documents=docs, question=input_text)
|
| 44 |
+
return response
|
|
|
|
| 45 |
|
| 46 |
+
iface = gr.Interface(fn=chatbot_interface, inputs="text", outputs="text")
|
| 47 |
+
iface.launch()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|