Update main.py

main.py

@@ -230,6 +230,60 @@ def chat():
         return jsonify({'error': str(e)}), 500
 
 
+@app.route('/chatgrouped', methods=['OPTIONS'])
+def options_grouped():
+    response = make_response()
+    response.headers.add("Access-Control-Allow-Origin", "*")
+    response.headers.add("Access-Control-Allow-Methods", "POST")
+    response.headers.add("Access-Control-Allow-Headers", "Content-Type, Authorization")
+    response.headers.add("Access-Control-Allow-Credentials", "true")
+    return response
+    
+@app.route('/chatgrouped', methods=['POST'])
+def chat_grouped():
+    try:
+        data = request.get_json()
+        messages = data.get('messages', [])
+
+        if messages:
+            results = []
+            # Update the model name to "text-davinci-003" (Ada)
+            prompt = "\n".join([f"{msg['role']}: {msg['content']}" for msg in messages])
+            response = openai.completions.create(
+                  model="gpt-3.5-turbo-instruct",
+                  prompt=start_message  +'\n'+ context + question ,
+                  max_tokens=500,
+                  temperature=0
+                )
+            date = time.ctime(time.time())
+            texte = prompt
+            data = Question(date, texte)
+            db.session.add(data)
+            db.session.commit()
+            question_id = data.id
+            resulta = response.choices[0].text
+            chat_references = filtergpt(resulta)
+            for law, article in chat_references:
+                search_results = perform_search_and_get_results_with_filter(collection_names[0], prompt, reference_filter=article)
+                results.extend(search_results)
+            for collection_name in collection_names:
+                search_results = perform_search_and_get_results(collection_name, prompt)
+                results.extend(search_results)
+            grouped_hits = {}
+            for i, hit in enumerate(results, 1):
+                second_number = hit.payload['numero_article']
+                if second_number not in grouped_hits:
+                    grouped_hits[second_number] = []
+                grouped_hits[second_number].append(hit)
+            return jsonify({'question': {'id': question_id, 'date': date, 'texte': texte},'result_qdrant':grouped_hits})
+        else:
+            return jsonify({'error': 'Invalid request'}), 400
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
+
+
+
 @app.route('/generateQuestions', methods=['OPTIONS'])
 def options_generate():
     response = make_response()