Adding Chatbot Feature

app.py

@@ -4,12 +4,23 @@ import mysql.connector
 import os
 import logging
 import random
+from google import genai
 
 
+
+def initialize_model(client,prompt):
+    chat = client.chats.create(
+    model="gemini-2.0-flash"
+    )
+    response = chat.send_message(prompt)
+
+    return chat
+
 app = Flask(__name__)
 CORS(app)
-logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
 logging.basicConfig(level=logging.INFO)
+client = genai.Client(api_key=os.getenv("GOOGLE_API_KEY"))
+chats = initialize_model(client,os.getenv("SYSTEM_PROMPT"))
 
 # === MySQL Connection ===
 def get_connection():
@@ -32,7 +43,7 @@ def root():
 def chat():
     data = request.get_json()
     message = data.get("message")
-    response = f"Echo: {message}"
+    response = chat.send_message(message)
     return jsonify({"response": response})
 
 

requirements.txt

@@ -5,4 +5,5 @@
  pydantic
  Flask
  Flask-Cors
- werkzeug
+ werkzeug
+ google-genai