from django import forms
from .models import Paciente, ArchivoPaciente, NotaMedica  # Agrega NotaMedica aquí

class PacienteForm(forms.ModelForm):
    class Meta:
        model = Paciente
        fields = ['nombre', 'edad', 'motivo_consulta', 'diagnostico', 'tratamiento_propuesto', 'tratamiento_realizado', 'observaciones']

class ArchivoPacienteForm(forms.ModelForm):
    class Meta:
        model = ArchivoPaciente
        fields = ['archivo', 'descripcion']

class NotaMedicaForm(forms.ModelForm):
    class Meta:
        model = NotaMedica
        fields = ['contenido']
        widgets = {
            'contenido': forms.Textarea(attrs={'rows': 4}),
        }