dhuynh95 commited on
Commit
e9c4839
1 Parent(s): c47f8c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -13
app.py CHANGED
@@ -54,21 +54,18 @@ with gr.Blocks() as demo:
54
  clear = gr.Button("Clear")
55
 
56
  def respond(openai_api_key, message, chat_history):
57
- openai.api_key = openai_api_key
58
  if not openai_api_key:
59
  return "No OpenAI key provided, please provide one.", chat_history
60
- try:
61
- index = load_index_from_storage(storage_context)
62
- query_engine = index.as_query_engine(similarity_top_k=3)
63
- response = query_engine.query(message)
64
- bot_message = response.response
65
- for i, row in get_podcast_and_youtube(response).iterrows():
66
- bot_message += f"\n\n\n Source: {row.podcast} \n\n Link: {row.youtube_id}"
67
- chat_history.append((message, bot_message))
68
- time.sleep(1)
69
- return "", chat_history
70
- except Exception as e:
71
- return f"Unexpected error raised: {e}", chat_history
72
 
73
  msg.submit(respond, [openai_api_key, msg, chatbot], [msg, chatbot])
74
  btn2.click(respond, [openai_api_key, msg, chatbot], [msg, chatbot])
 
54
  clear = gr.Button("Clear")
55
 
56
  def respond(openai_api_key, message, chat_history):
 
57
  if not openai_api_key:
58
  return "No OpenAI key provided, please provide one.", chat_history
59
+ openai.api_key = openai_api_key
60
+ index = load_index_from_storage(storage_context)
61
+ query_engine = index.as_query_engine(similarity_top_k=3)
62
+ response = query_engine.query(message)
63
+ bot_message = response.response
64
+ for i, row in get_podcast_and_youtube(response).iterrows():
65
+ bot_message += f"\n\n\n Source: {row.podcast} \n\n Link: {row.youtube_id}"
66
+ chat_history.append((message, bot_message))
67
+ time.sleep(1)
68
+ return "", chat_history
 
 
69
 
70
  msg.submit(respond, [openai_api_key, msg, chatbot], [msg, chatbot])
71
  btn2.click(respond, [openai_api_key, msg, chatbot], [msg, chatbot])