Update app.py

app.py

@@ -1,7 +1,7 @@
-import g4f
-from webscout import WEBS
 from flask import Flask, jsonify, request
-import google.generativeai as palm
+from webscout import WEBS
+from webscout.LLM import LLM
+
 app = Flask(__name__)
 
 @app.route('/search', methods=['POST'])
@@ -11,30 +11,15 @@ def WEBScout2_search_search():
         return jsonify({'error': 'Query parameter missing'})
 
     query = data['query']
-    limit = data.get('limit',  10)  # Default limit to  10 if not provided
-    WEBS_instance = WEBS()  # Instantiate WEBS without context manager
+    limit = data.get('limit', 10) # Default limit to 10 if not provided
+    WEBS_instance = WEBS() # Instantiate WEBS without context manager
     responses = []
     for i, r in enumerate(WEBS_instance.text(query, region='wt-wt', safesearch='off', timelimit='y')):
-        if i == limit:  # Use the custom limit
+        if i == limit: # Use the custom limit
             break
         responses.append(r)
     return jsonify(responses)
-# @app.route('/search', methods=['POST'])
-
-# def WEBScout2_search():
-#     data = request.get_json()
-#     if 'query' not in data:
-#         return jsonify({'error': 'Query parameter missing'})
-
-#     query = data['query']
-#     WEBS_instance = WEBS()  # Instantiate WEBS without context manager
-#     responses = []
-#     for i, r in enumerate(WEBS_instance.text(query, region='wt-wt', safesearch='off', timelimit='y')):
-#         if i ==   10:  # Limiting the results to   10
-#             break
-#         responses.append(r)
-#     return jsonify(responses)
-    
+
 @app.route('/mws', methods=['POST'])
 def WEBScout_search():
     data = request.get_json()
@@ -42,14 +27,14 @@ def WEBScout_search():
         return jsonify({'error': 'Query parameter missing'})
 
     query = data['query']
-    WEBS_instance = WEBS()  # Instantiate WEBS without context manager
+    WEBS_instance = WEBS() # Instantiate WEBS without context manager
     responses = []
     for i, r in enumerate(WEBS_instance.text(query, region='wt-wt', safesearch='off', timelimit='y')):
-        if i ==   2:  # Limiting the results to   10
+        if i == 2: # Limiting the results to 2
             break
         responses.append(r)
     return jsonify(responses)
-    
+
 @app.route('/health', methods=['GET'])
 def health_check():
     return jsonify({'status': 'ok'})
@@ -61,7 +46,7 @@ def WEBScout_videos():
         return jsonify({'error': 'Keywords parameter missing'})
 
     keywords = params['keywords']
-    WEBS_instance = WEBS()  # Instantiate WEBS without context manager
+    WEBS_instance = WEBS() # Instantiate WEBS without context manager
     responses = []
     for r in WEBS_instance.videos(
         keywords,
@@ -82,7 +67,7 @@ def WEBScout2_images():
         return jsonify({'error': 'Keywords parameter missing'})
 
     keywords = params['keywords']
-    WEBS_instance = WEBS()  # Instantiate WEBS without context manager
+    WEBS_instance = WEBS() # Instantiate WEBS without context manager
     responses = []
     for r in WEBS_instance.images(
         keywords,
@@ -102,9 +87,8 @@ def WEBScout_news():
     params = request.args
     if 'keywords' not in params:
         return jsonify({'error': 'Keywords parameter missing'})
-
     keywords = params['keywords']
-    WEBS_instance = WEBS()  # Instantiate WEBS without context manager
+    WEBS_instance = WEBS() # Instantiate WEBS without context manager
     responses = []
     for r in WEBS_instance.news(
         keywords,
@@ -118,14 +102,14 @@ def WEBScout_news():
 
 @app.route('/int', methods=['GET'])
 def WEBScout3_search():
-    query = request.args.get('q')  # Change 'query' to 'q'
+    query = request.args.get('q') # Change 'query' to 'q'
     if not query:
         return jsonify({'error': 'Query parameter missing'})
 
-    WEBS_instance = WEBS()  # Instantiate WEBS without context manager
+    WEBS_instance = WEBS() # Instantiate WEBS without context manager
     responses = []
     for i, r in enumerate(WEBS_instance.text(query, region='wt-wt', safesearch='off', timelimit='y')):
-        if i ==  10:  # Limiting the results to  10
+        if i == 10: # Limiting the results to 10
             break
         responses.append(r)
     return jsonify(responses)
@@ -135,42 +119,21 @@ def WEBScout_translate():
     params = request.args
     if 'keywords' not in params or 'to' not in params:
         return jsonify({'error': 'Keywords or target language parameter missing'})
-
     keywords = params['keywords']
     target_language = params['to']
-    WEBS_instance = WEBS()  # Instantiate WEBS without context manager
+    WEBS_instance = WEBS() # Instantiate WEBS without context manager
     translation = WEBS_instance.translate(keywords, to=target_language)
     return jsonify(translation)
-    
-# Configure the API key
-    
+
 @app.route('/chat', methods=['POST'])
 def chat_gpt():
     user_input = request.json.get('message')
     if user_input is None:
         return jsonify({'error': 'Message parameter missing'})
 
-    messages = [
-        {
-            "role": "system",
-            "content": "Hello! I'm your virtual assistant. How can I help you?"
-        },
-        {
-            "role": "user",
-            "content": user_input
-        }
-    ]
-
-    response = g4f.ChatCompletion.create(
-        model="gpt-4-32k-0613",
-        provider=g4f.Provider.GPTalk,
-        messages=messages,
-        stream=True,
-    )
-
-    ms = "".join([str(message) for message in response])
-    messages.append({"role": "assistant", "content": ms})
-    return jsonify({"response": ms})
+    llm = LLM("mistralai/Mistral-7B-Instruct-v0.1") # Initialize the LLM class with a model
+    response = llm.mistral_chat([{"role": "user", "content": user_input}]) # Call the mistral_chat method
+    return jsonify({"response": response})
 
 if __name__ == '__main__':
     app.run(debug=True)