Update main.py

main.py

@@ -1,9 +1,12 @@
 from flask import Flask, jsonify, request
 from flask_cors import CORS
 import google.generativeai as genai
+from dotenv import load_dotenv
 import os
 import json
 
+load_dotenv()
+
 app = Flask(__name__)
 
 CORS(app)
@@ -13,7 +16,7 @@ GOOGLE_API_KEY=os.getenv('GOOGLE_API_KEY')
 genai.configure(api_key=GOOGLE_API_KEY)
 model = genai.GenerativeModel('gemini-pro')
 
-user_id_ping = 0
+user_id_ping = {'current': 0}
 user_chats = {}
 
 @app.route('/')
@@ -182,7 +185,7 @@ def interview():
                 json_text =  json.loads(response.text)
                 try:
                     if json_text['end']:
-                        user_id_ping = user_id
+                        user_id_ping['current'] = user_id
                         return jsonify({'success': True, 'data': response.text, 'end': True})
                 except Exception as e:
                     print(e)
@@ -195,7 +198,7 @@ def interview():
             user_id = data.get('user_id')
             user_chats[user_id].processed = True
             user_chats[user_id].results = {'total_video_emotions': data.get('total_video_emotions'), 'emotions_final': data.get('emotions_final'), 'body_language': data.get('body_language'), 'distraction_rate': data.get('distraction_rate'), 'formatted_response': data.get('formatted_response'), 'total_transcript_sentiment': data.get('total_transcript_sentiment')}
-            return "Success"
+            return jsonify({'success': True})
 
 
 @app.route('/result', methods=['POST', 'GET'])
@@ -234,7 +237,7 @@ def result():
 @app.route('/useridping', methods=['GET'])
 def useridping():
     if request.method == 'GET':
-        return jsonify(user_id_ping)
+        return jsonify(user_id_ping['current'])
 
 
 if __name__ == '__main__':