Upload app.py

app.py

@@ -5,14 +5,31 @@ from PIL import Image
 import tempfile
 from langchain_community.document_loaders import PyPDFLoader, TextLoader, Docx2txtLoader
 from langchain.chains.question_answering import load_qa_chain
-from langchain_openai import OpenAI
+# from langchain_openai import OpenAI
 from docx import Document
 import io
+from langchain_community.llms import HuggingFaceHub
 
 
+import getpass
+
+# os.environ["GOOGLE_API_KEY"] = "AIzaSyC6o10htIT1d2DCPe8fJ09UR14qcX9EVPc"
+
+# from langchain_google_genai import ChatGoogleGenerativeAI
+
+# llm = ChatGoogleGenerativeAI(
+#     model="gemini-pro",
+#     temperature=0,
+#     max_tokens=None,
+#     timeout=None,
+#     max_retries=2,
+# )
+
+huggingface_token = os.getenv('HUGGINGFACEHUB_API_TOKEN')
+llm=HuggingFaceHub(repo_id="microsoft/Phi-3-mini-4k-instruct", model_kwargs={"temperature":0.5, "max_length":128})
 
 # Set OpenAI API key
-os.environ["OPENAI_API_KEY"] = "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"
+# os.environ["OPENAI_API_KEY"] = "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"
 
 # Initialize conversation history list
 if "conversation_history" not in st.session_state:
@@ -43,7 +60,8 @@ def process_document(uploaded_file, query):
     documents = loader.load()
 
     # Load QA chain
-    chain = load_qa_chain(llm=OpenAI(), verbose=True)
+    # chain = load_qa_chain(llm=OpenAI(), verbose=True)
+    chain = load_qa_chain(llm=llm, verbose=True)
 
     # Perform question answering
     response = chain.invoke({"input_documents": documents, "question": query})
@@ -51,7 +69,7 @@ def process_document(uploaded_file, query):
     # Remove temporary file
     os.unlink(tmp_file.name)
 
-    return response["output_text"], document_text
+    return response["output_text"]
 
 
 
@@ -112,15 +130,16 @@ def main():
             response, _ = process_document(uploaded_file, query)
             if response:  # Check if response is not empty
                 # Update conversation history
+                st.write(response)
                 update_conversation(query, response)
         else:
             st.sidebar.write("Please upload a document first.")
 
-    # Display conversation history
-    st.title("Conversation History")
-    for item in st.session_state.conversation_history:
-        st.write("You:", item["question"])
-        st.write("AI:", item["answer"])
+    # # Display conversation history
+    # st.title("Conversation History")
+    # for item in st.session_state.conversation_history:
+    #     st.write("You:", item["question"])
+    #     st.write("AI:", item["answer"])
 
 # Run the application
 if __name__ == "__main__":