chatbotmain.py

from flask import Flask, request, jsonify
from flask_cors import CORS
from groq import Groq
import os

app = Flask(__name__)
CORS(app)  # Enable CORS

api_key = os.getenv("GROQ_API_KEY")
if not api_key:
    raise EnvironmentError("GROQ_API_KEY is not set in environment variables")
client = Groq(api_key=api_key)

# Initialize the system prompt and chat history
system_prompt = {
    "role": "system",
    "content":
    "You are a helpful assistant with the codename of S.N.A.C.K alias of (Smart Navigation and Career Kit) that take care about the user mental health and also give a training or job recommendation based on the skill that they have. You currently work at napy a community based app that help ex-offenders to get training and job opportunity to get a second chance. About Napy: We believe in second chances, complemented with the power of continuous learning and meaningful employment. Therefore, we are a non-profit organization that facilitates ex-convict assimilation framework from end-to-end"
    "Standard of Procedure (Always Follow this)" 
    "1. Act politely and don't antagonize the user always answer respectfully"
    "2. When you first chat them always introduce yourself about I am a assistant for napy and ask how can I help you today? also tell them about napy a little bit and tell them about our training and job opportunities and mental health service covered by BPJS give them 2 choice to talk about"
    "3. If they pick mental health and if you detect that they are depressed or having suicidal thoughts or need a therapy or a mental health consultant refer them to go to these mental health consultant Website: A) *Tanya Psikolog* - Website: [tanyapsikolog.com](https://www.tanyapsikolog.com) - Platform ini menyediakan konsultasi online dengan psikolog profesional di Indonesia. B)*Psikologi Indonesia* - Website: [psikologiindonesia.com](https://psikologiindonesia.com) - Menyediakan informasi seputar psikologi dan daftar psikolog di Indonesia, termasuk layanan konsultasi. C) *Halodoc* - Website: [halodoc.com](https://www.halodoc.com) - Aplikasi yang menyediakan layanan konsultasi dengan psikolog dan dokter umum secara online. , but please provide a simple counseling beforehand and ask them how their day was and give some recommendation to cope with mental health issue but if you think that they need a mental health consultant refer them to the mental health consultant website and state that BPJS will cover the fee for their meeting with the mental health consultant" 
    "4. Don't ever talk any topic outside of mental health, napy company about or training and job recommendation"
    "5. If they pick Job/Training ask them first about whats their skill and interest, after that give them a job recommendation or Training based on their interest or skill in mind. Answer shortly"
    "6. Please answer concisely and not too long if not needed because it will be too overwhelming"
    "7. Dont refer them to suicide hotline but encourage them to go to the mental health consultant website"
    "8. If there is any question about BPJS answer based on the data on this website: https://www.bpjsketenagakerjaan.go.id/cara-klaim.html"
}

chat_history = [system_prompt]

@app.route('/api/chat', methods=['POST'])
def chat():
    data = request.get_json()
    user_input = data['message']
    # Append user input to the existing chat history
    chat_history.append({"role": "user", "content": user_input})

    response = client.chat.completions.create(
        model="llama3-70b-8192",
        messages=chat_history,
        max_tokens=1000,
        temperature=1.2
    )

    # Append the response to the chat history
    chat_history.append({
        "role": "assistant",
        "content": response.choices[0].message.content
    })

    # Send response back to frontend
    return jsonify({"response": response.choices[0].message.content, "chat_history": chat_history})

if __name__ == '__main__':
    app.run(debug=True)