from django import forms
from .models import Startup, Message, Profile

class StartupForm(forms.ModelForm):
    class Meta:
        model = Startup
        fields = ['name', 'headline', 'description', 'target_amount', 'status']
        widgets = {
            'name': forms.TextInput(attrs={'class': 'form-control', 'placeholder': 'Startup Name'}),
            'headline': forms.TextInput(attrs={'class': 'form-control', 'placeholder': 'A catchy one-liner'}),
            'description': forms.Textarea(attrs={'class': 'form-control', 'rows': 4, 'placeholder': 'What are you building?'}),
            'target_amount': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': 'Target funding amount (£)'}),
            'status': forms.Select(attrs={'class': 'form-select'}),
        }

class MessageForm(forms.ModelForm):
    class Meta:
        model = Message
        fields = ['content']
        widgets = {
            'content': forms.Textarea(attrs={'class': 'form-control', 'rows': 3, 'placeholder': 'Write your message...'}),
        }
