Update main.py

main.py

@@ -40,19 +40,7 @@ def llm(prompt, model):
     response = llm(prompt)
     return response
 
-
 def gpt_with_google_search(prompt, model):
-    search_results = search_google(prompt)
-    text = ""
-    ref = ""
-    for item in search_results:
-        text += item['title'] + "\n" + item['snippet'] + "\n\n"
-        ref += "- {} ({})\n".format(item['title'], item['link'])
-    results = llm(prompt = f'Summarize: {text}', model = model)
-    res = "{} \n\n {}".format(results, ref)
-    return res
-
-def gpt_with_google_search(prompt):
     search_results = search_google(prompt)
     text = ""
     ref = ""
@@ -61,7 +49,7 @@ def gpt_with_google_search(prompt):
         text += item['title'] + "\n" + item['snippet'] + "\n\n"
         ref += " [{}] {} ({})\n".format(ref_count, item['title'], item['link'])
         ref_count += 1
-    results = generate_text(f'Summarize: {text}')
+    results = llm(prompt = f'Summarize: {text}', model = model)
     res = "{} \n\n{}".format(results, ref)
     return res