First CVchat app working version

app.py

@@ -0,0 +1,63 @@
+import gradio as gr
+import logging
+from langchain.chains import RetrievalQA
+from langchain.llms import OpenAI
+from langchain.document_loaders import PDFMinerLoader
+from langchain.indexes import VectorstoreIndexCreator
+import os
+
+def set_openai_key(raw_key):
+    logging.warning(raw_key)
+    os.environ["OPENAI_API_KEY"] = raw_key
+
+
+def create_langchain(pdf_object):
+    logging.info(f"Creating langchain for {pdf_object.name}")
+    loader = PDFMinerLoader(pdf_object.name)
+    index_creator = VectorstoreIndexCreator()
+    docsearch = index_creator.from_loaders([loader])
+    chain = RetrievalQA.from_chain_type(
+        llm=OpenAI(),
+        chain_type="stuff",
+        retriever=docsearch.vectorstore.as_retriever(),
+        input_key="question",
+        verbose=True,
+        return_source_documents=True,
+    )
+    return chain, gr.Button.update(interactive=True)
+
+
+def ask_question(chain, question_text):
+    logging.info(type(chain))
+    return chain({"question": question_text})["result"]
+  
+def create_ask(pdf_object, question_text):
+    loader = PDFMinerLoader(pdf_object.name)
+    index_creator = VectorstoreIndexCreator()
+    docsearch = index_creator.from_loaders([loader])
+    chain = RetrievalQA.from_chain_type(
+        llm=OpenAI(),
+        chain_type="stuff",
+        retriever=docsearch.vectorstore.as_retriever(),
+        input_key="question",
+        verbose=True,
+        return_source_documents=True,
+    )
+    return chain({"question": question_text})["result"]
+
+
+with gr.Blocks() as demo:
+    # pdf_button = gr.Button(value="pdf_button")
+    oai_token = gr.Textbox(label="OpenAI Token", placeholder="Lm-iIas452gaw3erGtPar26gERGSA5RVkFJQST23WEG524EWEl")
+    oai_token.change(set_openai_key, oai_token)
+    pdf_object = gr.File(
+            label="Upload your CV in PDF format",
+            file_count="single",
+            type="file")
+    chain_state = gr.State()
+    # pdf_button.click(create_langchain, inputs=pdf_object, outputs=chain_state)
+    question_box = gr.Textbox(label="Question", value="Which are the candidate top hard skills? Enumerate in few bullet points.")
+    qa_button = gr.Button(value="Submit question", interactive=False)
+
+    lchain = pdf_object.change(create_langchain, inputs=pdf_object, outputs=[chain_state, qa_button])
+    qa_button.click(ask_question, inputs=[chain_state, question_box], outputs=gr.Textbox(label="Answer"))

requirements.txt

@@ -0,0 +1,6 @@
+openai
+gradio
+langchain
+pdfminer.six
+chromadb
+tiktoken