apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-05-01 03:46:25 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

SurfSense v0.0.5 beta

Browse source 
- Still need to add delete fuctions but what the hell
This commit is contained in:
DESKTOP-RTLN3BA\$punk 2024-11-11 00:25:28 -08:00
parent 1d67d0bf85
commit b3e2f9fc9c
8 changed files with 980 additions and 578 deletions
Download patch file Download diff file Expand all files Collapse all files

50
backend/models.py
View file

@ -1,7 +1,7 @@
from datetime import datetime
from typing import List
# from typing import List
from database import Base, engine
from sqlalchemy import Column, DateTime, ForeignKey, Integer, String, create_engine
from sqlalchemy import Column, DateTime, ForeignKey, Integer, String, Boolean, create_engine
from sqlalchemy.orm import relationship
class BaseModel(Base):
@ -18,8 +18,8 @@ class Chat(BaseModel):
title = Column(String)
chats_list = Column(String)
user_id = Column(ForeignKey('users.id'))
user = relationship('User')
search_space_id = Column(Integer, ForeignKey('searchspaces.id'))
search_space = relationship('SearchSpace', back_populates='chats')
class Documents(BaseModel):
@ -31,30 +31,50 @@ class Documents(BaseModel):
file_type = Column(String)
document_metadata = Column(String)
page_content = Column(String)
desc_vector_start = Column(Integer, default=0)
desc_vector_end = Column(Integer, default=0)
search_space_id = Column(ForeignKey('searchspaces.id'))
search_space = relationship('SearchSpace')
summary_vector_id = Column(String)
search_space_id = Column(Integer, ForeignKey("searchspaces.id"))
search_space = relationship("SearchSpace", back_populates="documents")
class Podcast(BaseModel):
__tablename__ = "podcasts"
title = Column(String)
created_at = Column(DateTime, default=datetime.now)
is_generated = Column(Boolean, default=False)
podcast_content = Column(String, default="")
file_location = Column(String, default="")
search_space_id = Column(Integer, ForeignKey("searchspaces.id"))
search_space = relationship("SearchSpace", back_populates="podcasts")
user_id = Column(ForeignKey('users.id'))
user = relationship('User')
class SearchSpace(BaseModel):
__tablename__ = "searchspaces"
search_space = Column(String, unique=True)
name = Column(String, index=True)
description = Column(String)
created_at = Column(DateTime, default=datetime.now)
documents = relationship(Documents)
user_id = Column(Integer, ForeignKey("users.id"))
user = relationship("User", back_populates="search_spaces")
documents = relationship("Documents", back_populates="search_space", order_by="Documents.id")
podcasts = relationship("Podcast", back_populates="search_space", order_by="Podcast.id")
chats = relationship('Chat', back_populates='search_space', order_by='Chat.id')
class User(BaseModel):
__tablename__ = "users"
username = Column(String, unique=True, index=True)
hashed_password = Column(String)
chats = relationship(Chat, order_by="Chat.id")
documents = relationship(Documents, order_by="Documents.id")
search_spaces = relationship("SearchSpace", back_populates="user")
# Create the database tables if they don't exist
User.metadata.create_all(bind=engine)
User.metadata.create_all(bind=engine)