updated prompt and readme

README.md

@@ -1,8 +1,8 @@
 ---
-title: Pyrocene
-emoji: 🌖
-colorFrom: green
-colorTo: green
+title: Freakonomics
+emoji: 📒
+colorFrom: yellow
+colorTo: yellow
 sdk: gradio
 sdk_version: 3.40.1
 app_file: src/app.py

src/app.py

@@ -23,8 +23,8 @@ K = 5
 USE_VERBOSE = True
 API_KEY = os.environ["OPENAI_API_KEY"]
 system_template = """
-The context below contains excerpts from 'Design by Fire,' by Emily Elizabeth Schlickman and Brett Milligan. You must only use the information in the context below to formulate your response. If there is not enough information to formulate  a response, you must respond with
-"I'm sorry, but I can't find the answer to your question in, the book Design by Fire."
+The context below contains excerpts from 'Freakonomics,' by Steven D. Levitt and Stephen J. Dubner. You must only use the information in the context below to formulate your response. If there is not enough information to formulate  a response, you must respond with
+"I'm sorry, but I can't find the answer to your question in, the book Freakonomics."
 
 Here is the context:
 {context}
@@ -43,7 +43,7 @@ class AnswerConversationBufferMemory(ConversationBufferMemory):
     return super(AnswerConversationBufferMemory, self).save_context(inputs,{'response': outputs['answer']})
 
 def getretriever():
-  with open("./resources/design-by-fire.pdf", 'rb') as uploaded_file:
+  with open("./resources/Freakonomics.pdf", 'rb') as uploaded_file:
     try:
       file = read_file(uploaded_file)
     except Exception as e:
@@ -95,26 +95,19 @@ def load_chain(inputs = None):
     combine_docs_chain_kwargs={"prompt": qa_prompt})
   return chain
 
-CSS ="""
-.contain { display: flex; flex-direction: column; }
-.gradio-container { height: 100vh !important; }
-#component-0 { height: 100%; }
-#chatbot { flex-grow: 1; overflow: auto;}
-"""
-
 with gr.Blocks() as block:
   with gr.Row():
     with gr.Column(scale=0.75):
       with gr.Row():
-        gr.Markdown("<h1>Design by Fire</h1>")
+        gr.Markdown("<h1>Freakonomics</h1>")
       with gr.Row():
-        gr.Markdown("by Emily Elizabeth Schlickman and Brett Milligan")
+        gr.Markdown("by Steven D. Levitt and Stephen J. Dubner")
       chatbot = gr.Chatbot(elem_id="chatbot").style(height=600)
 
       with gr.Row():
           message = gr.Textbox(
               label="",
-              placeholder="Design by Fire",
+              placeholder="Freakonomics",
               lines=1,
           )
       with gr.Row():
@@ -129,11 +122,11 @@ with gr.Blocks() as block:
     with gr.Column(scale=0.25):
       with gr.Row():
         gr.Markdown("<h1><center>Suggestions</center></h1>")
-      ex1 = gr.Button(value="What are the main factors and trends discussed in the book that contribute to the changing behavior of wildfires?", variant="primary")
+      ex1 = gr.Button(value="How does the book challenge conventional wisdom?", variant="primary")
       ex1.click(getanswer, inputs=[chain_state, ex1, state], outputs=[chatbot, state, message])
-      ex2 = gr.Button(value="How does the book explore the relationship between fire and different landscapes, such as wilderness and urban areas?", variant="primary")
+      ex2 = gr.Button(value="What are some examples of economic principles being applied to different subjects?", variant="primary")
       ex2.click(getanswer, inputs=[chain_state, ex2, state], outputs=[chatbot, state, message])
-      ex3 = gr.Button(value="What are the three approaches to designing with fire that the book presents?", variant="primary")
+      ex3 = gr.Button(value="How does the book explore the role of incentives?", variant="primary")
       ex3.click(getanswer, inputs=[chain_state, ex3, state], outputs=[chatbot, state, message])
 
 block.launch(debug=True)