Update app.py

app.py

@@ -1,50 +1,52 @@
 from flask import Flask, jsonify, request
+import os
 from webscout import WEBS
 from webscout.LLM import LLM
 
 app = Flask(__name__)
 
+# Create a directory for the .cache if it doesn't exist
+cache_dir = os.path.join(os.path.expanduser("~"), ".cache")
+if not os.path.exists(cache_dir):
+    os.makedirs(cache_dir)
+
 @app.route('/search', methods=['POST'])
 def WEBScout2_search_search():
     data = request.get_json()
     if 'query' not in data:
         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')):
         responses.append(r)
     return jsonify(responses)
-    
+
 @app.route('/int', methods=['GET'])
 def WEBScout3_search_search_search_search():
     query = request.args.get('q')
     if not query:
         return jsonify({'error': 'Query parameter missing'})
-
-    limit = request.args.get('limit', default=10, type=int) # Default limit to 10 if not provided
-    WEBS_instance = WEBS() # Instantiate WEBS without context manager
+    limit = request.args.get('limit', default=10, type=int)  # 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('/mws', methods=['POST'])
 def WEBScout_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
+    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 2
+        if i == 2:  # Limiting the results to 2
             break
         responses.append(r)
     return jsonify(responses)
@@ -58,9 +60,8 @@ def WEBScout_videos():
     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.videos(
         keywords,
@@ -79,9 +80,8 @@ def WEBScout2_images():
     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.images(
         keywords,
@@ -102,7 +102,7 @@ def WEBScout_news():
     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,
@@ -114,7 +114,6 @@ def WEBScout_news():
         responses.append(r)
     return jsonify(responses)
 
-
 @app.route('/translate', methods=['GET'])
 def WEBScout_translate():
     params = request.args
@@ -122,7 +121,7 @@ def WEBScout_translate():
         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)
 
@@ -131,9 +130,8 @@ def chat_gpt():
     user_input = request.json.get('message')
     if user_input is None:
         return jsonify({'error': 'Message parameter missing'})
-
-    llm = LLM("mistralai/Mixtral-8x7B-Instruct-v0.1") # Initialize the LLM class with a model
-    response = llm.mistral_chat([{"role": "user", "content": user_input}]) # Call the mistral_chat method
+    llm = LLM("mistralai/Mixtral-8x7B-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__':