Update app.py

app.py

@@ -10,52 +10,31 @@ from dotenv import load_dotenv
 load_dotenv()
 # Replace with your key
 client = OpenAI()
-you_key = os.getenv("YOU_API_KEY")
 username = os.getenv("USERNAME")
 password = os.getenv('PASSWORD')
 
-def get_ai_snippets_for_query(query):
-    headers = {"X-API-Key": you_key}
-    params = {"query": query}
-    return requests.get(
-        f"https://api.ydc-index.io/search?query={query}",
-        params=params,
-        headers=headers,
-    ).json().get('hits')
-
-
-def get_web_search_you(query):
-    docs = get_ai_snippets_for_query(query)
-    markdown = ""
-    for doc in docs:
-        for key, value in doc.items():
-            if key == 'snippets':
-                markdown += f"{key}:\n"
-                for snippet in value:
-                    markdown += f"- {snippet}\n"
-            else:
-                markdown += f"{key}: {value}\n"
-        markdown += "\n"
-    return markdown
-
 
 def predict(message, history, _n_web_search, _strategy):
     # docs = get_web_search_you(message)
 
-    docs = get_docs_from_web(message, history[-1:], _n_web_search, _strategy)
     partial_message = ''
     information = ''
-    for doc in docs:
-        if isinstance(doc, dict):
-            information = doc.get('data')
-        else:
-            partial_message = partial_message + doc
-            yield partial_message
+
+    if _web_search_enabled == 'On':
+        docs = get_docs_from_web(message, history[-1:], _n_web_search, _strategy)
+        for doc in docs:
+            if isinstance(doc, dict):
+                information = doc.get('data')
+            else:
+                partial_message = partial_message + doc
+                yield partial_message
+                
     system_prompt = """
             You are an advanced chatbot.
             Today's date - {date}
 
-   When answering a question, adhere to the following revised rules:
+When answering a question, adhere to the following revised rules:
+- Use the "Information for reference" data to answer the question. If no answer found in "Information for reference" data then try to give answer.
 - The "Information for reference" data is provided in the chunks with each chunk having its own source as url.
 - Generate human-like text in response to input, reflecting your status as a sophisticated language model.
 - Abstain from offering any health or medical advice and ensure all responses maintain this guideline strictly.
@@ -106,6 +85,10 @@ n_web_search = gr.Slider(1, 10, value=3, step=1, label="Web searches",
 strategy = gr.Radio(["Deep", "Normal", "Normal Fast"], label="Strategy", value="Normal",
                     info="Select web search analysis type. Please keep in mind that deep analysis will take more time than normal analysis.")
 
-app = gr.ChatInterface(predict, additional_inputs=[n_web_search, strategy])
+web_search_enabled = gr.Radio(["On", "Off"], label="Web Search", value="Off",
+                              info="Select web search option on and off.")
+
+app = gr.ChatInterface(predict, additional_inputs=[n_web_search, strategy, web_search_enabled])
+
 app.queue(default_concurrency_limit=5)
 app.launch(debug=True, share=False, auth=(username, password), auth_message='Twimbit project ABM')