henryhyunwookim commited on
Commit
b735389
1 Parent(s): 6c3ea70

Update app.py

Browse files

update function and variable names

Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -3,7 +3,7 @@ import serpapi
3
  import gradio as gr
4
  from langchain_google_genai import ChatGoogleGenerativeAI
5
 
6
- from utils.utils import summarize_search_result, get_prompt
7
 
8
 
9
  def main(search_query, search_location, serp_api_key):
@@ -13,9 +13,9 @@ def main(search_query, search_location, serp_api_key):
13
  api_key=serp_api_key
14
  ) # engine="google", location="Austin, Texas", hl="en", gl="us"
15
 
16
- search_summary = summarize_search_result(search_response)
17
 
18
- prompt = get_prompt(search_query, search_summary)
19
 
20
  chat_model = ChatGoogleGenerativeAI(
21
  model="gemini-pro",
 
3
  import gradio as gr
4
  from langchain_google_genai import ChatGoogleGenerativeAI
5
 
6
+ from utils.utils import get_top_results, get_prompt
7
 
8
 
9
  def main(search_query, search_location, serp_api_key):
 
13
  api_key=serp_api_key
14
  ) # engine="google", location="Austin, Texas", hl="en", gl="us"
15
 
16
+ top_results = get_top_results(search_response)
17
 
18
+ prompt = get_prompt(search_query, top_results)
19
 
20
  chat_model = ChatGoogleGenerativeAI(
21
  model="gemini-pro",