themeetjani commited on
Commit
01f902f
1 Parent(s): 18d1238

Update pages/AI_Chatbot.py

Browse files
Files changed (1) hide show
  1. pages/AI_Chatbot.py +13 -3
pages/AI_Chatbot.py CHANGED
@@ -2,17 +2,27 @@ import openai
2
  import os
3
  import streamlit as st
4
  from streamlit import session_state
5
- openai.api_key = os.getenv("OPENAI_API_KEY")
6
- from langchain.document_loaders import PyPDFLoader
 
7
  from langchain.document_loaders import WebBaseLoader, PyPDFLoader, TextLoader
8
  from langchain.indexes import VectorstoreIndexCreator
9
  from langchain.embeddings import HuggingFaceEmbeddings
10
  from langchain.docstore.document import Document
 
 
 
11
 
12
  st.title("Chat with data")
 
13
 
14
  uploaded_file = st.file_uploader("Choose a file")
15
- uploaded_file = uploaded_file.name
 
 
 
 
 
16
  def extract(uploaded_file):
17
  res = []
18
  loader = PyPDFLoader(uploaded_file)
 
2
  import os
3
  import streamlit as st
4
  from streamlit import session_state
5
+ import base64
6
+ import tempfile
7
+ from pathlib import Path
8
  from langchain.document_loaders import WebBaseLoader, PyPDFLoader, TextLoader
9
  from langchain.indexes import VectorstoreIndexCreator
10
  from langchain.embeddings import HuggingFaceEmbeddings
11
  from langchain.docstore.document import Document
12
+ openai.api_key = os.getenv("OPENAI_API_KEY")
13
+ from langchain.document_loaders import PyPDFLoader
14
+ from langchain.chat_models import ChatOpenAI
15
 
16
  st.title("Chat with data")
17
+ model = ChatOpenAI(model = 'gpt-4', max_tokens = 100,temperature=0)
18
 
19
  uploaded_file = st.file_uploader("Choose a file")
20
+ if uploaded_file is not None:
21
+ # Make temp file path from uploaded file
22
+ with tempfile.NamedTemporaryFile(delete=False) as tmp_file:
23
+ fp = Path(tmp_file.name)
24
+ fp.write_bytes(uploaded_file.getvalue())
25
+ print(tmp_file.name,"path")
26
  def extract(uploaded_file):
27
  res = []
28
  loader = PyPDFLoader(uploaded_file)