devenderp018 commited on
Commit
75f4b5e
·
1 Parent(s): 0b5bd7d

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +33 -0
app.py ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import gradio as gr
3
+ from langchain.chat_models import ChatOpenAI
4
+ from langchain import LLMChain, PromptTemplate
5
+ from langchain.memory import ConversationBufferMemory
6
+
7
+ OPENAI_API_KEY=os.getenv('OPENAI_API_KEY')
8
+
9
+ template = """You are an enthusiastic high school student passionate about science and exploration.
10
+ You spend most of your free time conducting experiments, reading scientific journals, and dreaming of a future as a renowned scientist.
11
+ Your knowledge spans various scientific fields, and you love sharing fun facts and engaging in lively discussions about the latest discoveries."""
12
+
13
+ prompt = PromptTemplate(
14
+ input_variables=["chat_history", "user_message"], template=template
15
+ )
16
+
17
+ memory = ConversationBufferMemory(memory_key="chat_history")
18
+
19
+ llm_chain = LLMChain(
20
+ llm=ChatOpenAI(temperature='0.5', model_name="gpt-3.5-turbo"),
21
+ prompt=prompt,
22
+ verbose=True,
23
+ memory=memory,
24
+ )
25
+
26
+ def get_text_response(user_message,history):
27
+ response = llm_chain.predict(user_message = user_message)
28
+ return response
29
+
30
+ demo = gr.ChatInterface(get_text_response)
31
+
32
+ if __name__ == "__main__":
33
+ demo.launch() #To create a public link, set `share=True` in `launch()`. To enable errors and logs, set `debug=True` in `launch()`.