|
from app.db import Base |
|
from sqlalchemy import Column, Float, ForeignKey, Integer, String |
|
from sqlalchemy.orm import relationship |
|
|
|
|
|
class Feedback(Base): |
|
__tablename__ = "feedbacks" |
|
|
|
id = Column(Integer, primary_key=True, index=True) |
|
score = Column(Float) |
|
predicted_class = Column(String(50)) |
|
feedback = Column(String(255)) |
|
user_id = Column(Integer, ForeignKey("users.id")) |
|
image_file_name = Column(String(255)) |
|
user = relationship("User", back_populates="feedbacks") |
|
|
|
def __init__( |
|
self, score, predicted_class, feedback, image_file_name, user, *args, **kwargs |
|
): |
|
self.predicted_class = predicted_class |
|
self.feedback = feedback |
|
self.score = score |
|
self.image_file_name = image_file_name |
|
self.user = user |
|
|