Update app.py

app.py

@@ -1,34 +1,27 @@
 from huggingface_hub import InferenceClient
 import gradio as gr
 import requests
+import json
 
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
-def search_url(search_query):
-    API_KEY = st.secrets['AIzaSyDseKKQCAUBmPidu_QapnpJCGLueDWYJbE']
-    SEARCH_ENGINE_ID = st.secrets['001ae9bf840514e61']
+def google_search(query, **kwargs):
+    api_key = 'AIzaSyDseKKQCAUBmPidu_QapnpJCGLueDWYJbE'
+    cse_id = '001ae9bf840514e61'
     
-    url = 'https://customsearch.googleapis.com/customsearch/v1'
-
+    service_url = 'https://www.googleapis.com/customsearch/v1'
     params = {
-        'q': search_query,
-        'key': API_KEY,
-        'cx': SEARCH_ENGINE_ID,
+        'key': api_key,
+        'cx': cse_id,
+        'q': query,
+        **kwargs
     }
-
-    response = requests.get(url, params=params)
-
-    results = response.json()
-
-    # print(results)
-
-    if 'items' in results:
-        for i in range(min(5, len(results['items']))):
-            print(f"Link {i + 1}: {results['items'][i]['link']}")
-        return results['items'][:5]
+    response = requests.get(service_url, params=params)
+    if response.status_code == 200:
+        return json.loads(response.text)['items']
     else:
-        print("No search results found.")
-        return None
+        print(f'Error: {response.status_code}')
+        return []
         
 def tokenize(text):
     return text
@@ -69,13 +62,13 @@ def generate(prompt, history, system_prompt, temperature=0.2, max_new_tokens=512
 
 def generateS(prompt, history, system_prompt, temperature=0.2, max_new_tokens=512, top_p=0.95, repetition_penalty=1.0):
     
-    stream = search_url(prompt)
-    # output = ""
+    stream = google_search(prompt)
+    output = ""
 
-    # for response in stream:
-    #     output += response.token.text
-    #     yield output
-    return stream
+    for response in stream:
+        output += response.token.text
+        yield output
+    return output
     
 additional_inputs=[
     gr.Textbox(
@@ -124,7 +117,7 @@ additional_inputs=[
 mychatbot = gr.Chatbot(
     avatar_images=["./user.png", "./botm.png"], bubble_full_width=False, show_label=False, show_copy_button=True, likeable=False)
 
-demo = gr.ChatInterface(fn=generate, 
+demo = gr.ChatInterface(fn=generateS, 
                         chatbot=mychatbot,
                         additional_inputs=additional_inputs,
                         title="Kamran's Mixtral 8x7b Chat",