orrinin commited on
Commit
1615c34
·
verified ·
1 Parent(s): fe68312

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -7
app.py CHANGED
@@ -105,7 +105,7 @@ def infer(message:str, history: list):
105
  documents, option = extract_web(message["text"])
106
  db_path = create_col(documents)
107
  elif not message["text"].startswith("http://") and not message["text"].startswith("https://") and len(history) == 0:
108
- return gr.Error("Please input an url or upload file at first.")
109
 
110
 
111
  # Load from disk
@@ -123,7 +123,7 @@ def infer(message:str, history: list):
123
  )
124
 
125
  if option == "web" and len(history) == 0:
126
- response = "Get the web data! You can ask it."
127
  else:
128
  question = message['text']
129
 
@@ -131,10 +131,11 @@ def infer(message:str, history: list):
131
  index.as_retriever(),
132
  memory=memory,
133
  context_prompt=(
134
- "You are a chatbot, able to have normal interactions, as well as talk"
135
- " about the Kendrick and Drake beef."
136
- "Here are the relevant documents for the context:\n"
137
  "{context_str}"
 
 
138
  "\nInstruction: Use the previous chat history, or the context above, to interact and help the user."
139
  ),
140
  verbose=True,
@@ -155,9 +156,9 @@ def infer(message:str, history: list):
155
 
156
 
157
 
158
- chatbot = gr.Chatbot()
159
 
160
- with gr.Blocks(theme="soft") as demo:
161
  gr.ChatInterface(
162
  fn = infer,
163
  title = "RAG demo",
 
105
  documents, option = extract_web(message["text"])
106
  db_path = create_col(documents)
107
  elif not message["text"].startswith("http://") and not message["text"].startswith("https://") and len(history) == 0:
108
+ raise gr.Error("请先输入网址或上传文档。")
109
 
110
 
111
  # Load from disk
 
123
  )
124
 
125
  if option == "web" and len(history) == 0:
126
+ response = "获取成功,开始对话吧!"
127
  else:
128
  question = message['text']
129
 
 
131
  index.as_retriever(),
132
  memory=memory,
133
  context_prompt=(
134
+ "You are an assistant for question-answering tasks."
135
+ "Use the following context to answer the question:\n"
 
136
  "{context_str}"
137
+ "\nIf you don't know the answer, just say that you don't know."
138
+ "Use five sentences maximum and keep the answer concise, all response in Chinese."
139
  "\nInstruction: Use the previous chat history, or the context above, to interact and help the user."
140
  ),
141
  verbose=True,
 
156
 
157
 
158
 
159
+ chatbot = gr.Chatbot(placeholder="请先输入网址或上传文档<br>然后进行对话")
160
 
161
+ with gr.Blocks(theme="soft", fill_height="true") as demo:
162
  gr.ChatInterface(
163
  fn = infer,
164
  title = "RAG demo",