Spaces:
Sleeping
Sleeping
File size: 5,270 Bytes
335f242 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 |
import streamlit as st
import sqlite3
from mistralai import Mistral
from datetime import datetime
import logging
import platform
from dotenv import load_dotenv
import os
# Configuration du logging
logging.basicConfig(level=logging.INFO, handlers=[logging.StreamHandler()])
logger = logging.getLogger(__name__)
if platform.system() == "Windows":
# Specify the path to your .env file
dotenv_path = os.path.join(os.path.dirname(__file__), '..', '..', '.env')
# Load the .env file
load_dotenv(dotenv_path)
else:
load_dotenv()
# Establish PostgreSQL connection
# try:
# db = psycopg2.connect(
# host=os.environ.get("POSTGRES_HOST"),
# user=os.environ.get("POSTGRES_USER"),
# password=os.environ.get("POSTGRES_PASSWORD"),
# dbname=os.environ.get("POSTGRES_DBNAME"),
# port=os.environ.get("POSTGRES_PORT")
# )
# except psycopg2.OperationalError as err:
# logger.error(f"Operational error: {err}")
# except psycopg2.Error as err:
# logger.error(f"Database connection error: {err}")
# Configuration de la base de données SQLite
DB_NAME = "chat_history.db"
conn = sqlite3.connect(DB_NAME, check_same_thread=False)
# Initialisation du client Mistral
# api_key = "RXjfbTO7wkOU0RwrwP7XpFfcj1K5eq40"
# api_key = os.environ.get("MISTRAL_API_KEY")
# mistral_client = Mistral(api_key=api_key)
def get_cursor():
"""
Validate and obtain a cursor for database operations.
Returns:
cursor: A database cursor.
"""
try:
return conn.cursor()
except sqlite3.Error as err:
logger.error(f"Error obtaining cursor: {err}")
return None
def init_db():
cursor = get_cursor()
try:
cursor.execute('''
CREATE TABLE IF NOT EXISTS conversations (
id INTEGER PRIMARY KEY AUTOINCREMENT,
created_at TEXT NOT NULL,
title TEXT NOT NULL
)
''')
cursor.execute('''
CREATE TABLE IF NOT EXISTS messages (
id INTEGER PRIMARY KEY AUTOINCREMENT,
conversation_id INTEGER NOT NULL,
role TEXT NOT NULL,
content TEXT NOT NULL,
timestamp TEXT NOT NULL,
FOREIGN KEY (conversation_id) REFERENCES conversations (id)
)
''')
return
except sqlite3.Error as err:
logger.error(f"Error while connecting to database: {err}")
return
raise HTTPException(status_code=500, detail=str(err))
finally:
conn.commit()
# conn.close()
def save_message(conversation_id, role, content):
cursor = get_cursor()
try:
timestamp = datetime.now().strftime("%d/%m/%Y %H:%M:%S")
cursor.execute("INSERT INTO messages (conversation_id, role, content, timestamp) VALUES (?, ?, ?, ?)",
(conversation_id, role, content, timestamp))
except sqlite3.Error as err:
logger.error(f"Error while connecting to database: {err}")
return
raise HTTPException(status_code=500, detail=str(err))
finally:
conn.commit()
# conn.close()
def create_conversation(title):
cursor = get_cursor()
try:
created_at = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
cursor.execute("INSERT INTO conversations (created_at, title) VALUES (?, ?)", (created_at, title))
conversation_id = cursor.lastrowid
return conversation_id
except sqlite3.Error as err:
logger.error(f"Error while connecting to database: {err}")
raise Exception(status_code=500, detail=str(err))
finally:
conn.commit()
# conn.close()
def load_messages(conversation_id):
cursor = get_cursor()
try:
cursor.execute("SELECT role, content FROM messages WHERE conversation_id = ? ORDER BY timestamp ASC", (conversation_id,))
data = [{"role": row[0], "content": row[1]} for row in cursor.fetchall()]
return data
except sqlite3.Error as err:
logger.error(f"Error while connecting to database: {err}")
raise Exception(status_code=500, detail=str(err))
finally:
conn.commit()
# conn.close()
def load_conversations():
cursor = get_cursor()
try:
cursor.execute("SELECT * FROM conversations ORDER BY created_at DESC")
data = cursor.fetchall()
return data
except sqlite3.Error as err:
logger.error(f"Error while connecting to database: {err}")
raise Exception(status_code=500, detail=str(err))
finally:
conn.commit()
# conn.close()
def update_conversation(conversation_id):
cursor = get_cursor()
try:
new_timer = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
cursor.execute("UPDATE conversations SET created_at = ? WHERE id = ?",(new_timer, conversation_id))
except sqlite3.Error as err:
logger.error(f"Error while connecting to database: {err}")
raise Exception(status_code=500, detail=str(err))
finally:
conn.commit()
# conn.close() |