{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 22,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "JA\n"
     ]
    }
   ],
   "source": [
    "from openai import OpenAI\n",
    "from dotenv import load_dotenv\n",
    "load_dotenv()\n",
    "\n",
    "history = []\n",
    "\n",
    "system_prompt = \"\"\"Du ska bedömma frågor och svara endast JA eller NEJ.\n",
    "\n",
    "Om frågan är relaterad till 1177 sjukrådgivningsdatabas, hälsa,vård, välmående, sjukrådgivning ska du svara \"JA\".\n",
    "\n",
    "Om frågan inte är relaterad till ovanämnde ämnen ska du svara NEJ\"\"\"\n",
    "\n",
    "\n",
    "fråga = \"Hur kan jag lindra min pollenallergi?\"\n",
    "\n",
    "user_prompt = f\"\"\"Svara endast JA eller NEJ ifall följande fråga är är relevant till 1177, sjukrådgivning, hälsa, vård och välmående\n",
    "\n",
    "Frågan är: {fråga}\"\"\"\n",
    "\n",
    "\n",
    "history.append({\"role\": \"system\", \"content\": system_prompt})\n",
    "history.append({\"role\": \"user\", \"content\": user_prompt})\n",
    "\n",
    "\n",
    "client = OpenAI()\n",
    "response = client.chat.completions.create(\n",
    "        model='gpt-3.5-turbo',\n",
    "        messages=history,\n",
    "        temperature=0.4,\n",
    "        stream=False\n",
    "    )\n",
    "\n",
    "print(response.choices[0].message.content)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Svara endast JA eller NEJ ifall följande fråga är är relevant till 1177, sjukrådgivning, hälsa, vård och välmående\n",
      "\n",
      "Frågan är: Hur kan jag förbättra min sömn?\n"
     ]
    }
   ],
   "source": [
    "print(user_prompt)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "response = client.chat.completions.create(\n",
    "        model='gpt-3.5-turbo',\n",
    "        messages=history,\n",
    "        temperature=0.4,\n",
    "        stream=True\n",
    "    )"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 29,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "NEJ\n"
     ]
    }
   ],
   "source": [
    "from RAG_class import RAG_1177\n",
    "rag = RAG_1177()\n",
    "\n",
    "fråga = \"Hur lagar jag en banankaka?\"\n",
    "svar = rag.relevant_question(fråga)\n",
    "\n",
    "print(svar)\n",
    "\n"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.11.9"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}