Update app.py

app.py

@@ -1,8 +1,7 @@
 import streamlit as st
-from langchain.llms import OpenAI, Cohere
+from langchain.llms import CTransformers
 import os
-from langchain.llms import OpenAI
-from langchain.embeddings import OpenAIEmbeddings,CohereEmbeddings
+from langchain.embeddings import HuggingFaceEmbeddings
 from langchain.document_loaders import PyPDFLoader
 from langchain.vectorstores import Chroma
 from langchain.agents.agent_toolkits import (
@@ -17,12 +16,10 @@ from langchain.agents import Tool
 from langchain.tools import DuckDuckGoSearchRun
 # from langchain.utilities import WikipediaAPIWrapper
 from langchain.chains import RetrievalQA
-from langchain.chat_models import ChatOpenAI
 from langchain.agents import Tool
 from langchain.agents import initialize_agent
 from langchain.agents import AgentType
 from langchain.memory import ConversationBufferMemory
-from langchain.chat_models import ChatOpenAI
 from langchain.prompts import ChatPromptTemplate
 from typing import Dict, Union, Any, List
 
@@ -30,7 +27,6 @@ from langchain.callbacks.base import BaseCallbackHandler
 from langchain.schema import AgentAction
 from langchain.agents import AgentType, initialize_agent, load_tools
 from langchain.callbacks import tracing_enabled
-from langchain.llms import OpenAI
 from langchain.chat_models import ChatOpenAI
 from langchain import PromptTemplate
 from fpdf import FPDF
@@ -40,18 +36,18 @@ from fpdf import FPDF
 st.title('MedTechAI')
 
 persist_directory = ""
-model = ""
-with st.sidebar:
-    with st.form('Cohere/OpenAI'):
-        mod = st.radio('Choose OpenAI/Cohere', ('OpenAI', 'Cohere'))
-        api_key = st.text_input('Enter API key', type="password")
-        # model = st.radio('Choose Company', ('ArtisanAppetite foods', 'BMW','Titan Watches'))
-        submitted = st.form_submit_button("Submit")
-
 
 
 embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2", model_kwargs={'device': 'cpu'})
 
+print("""
+Duplicated space from WeekendAI/MedicalBot
+https://huggingface.co/spaces/WeekendAI/MedicalBot
+
+This version has been disconnected from OpenAI and needs no API key.
+This uses the TheBloke/Llama-2-7B-Chat-GGML model and the sentence-transformers/all-MiniLM-L6-v2 HuggingFaceEmbeddings
+
+""")
 
 def report(Symptoms,history,age,gen,response):
     prompt_updates=f"""
@@ -100,9 +96,6 @@ def report(Symptoms,history,age,gen,response):
     ```
     """
     print("report")
-    model_name = "gpt-3.5-turbo-16k"
-    temperature = 0.0
-    #llm = OpenAI(model_name=model_name, temperature=temperature)
     llm = CTransformers(model = "TheBloke/Llama-2-7B-Chat-GGML",model_type="llama",max_new_tokens = 2048,temperature = 0.5)
     out = llm(prompt_updates)