from django import forms
from .models import SectorApplication
from credit_scoring.models import CreditApplication

class CreditApplicationForm(forms.ModelForm):
    class Meta:
        model = CreditApplication
        fields = ['business_name', 'registration_number', 'amount_requested', 'purpose_of_funding']
        widgets = {
            'business_name': forms.TextInput(attrs={'class': 'form-control'}),
            'registration_number': forms.TextInput(attrs={'class': 'form-control'}),
            'amount_requested': forms.NumberInput(attrs={'class': 'form-control'}),
            'purpose_of_funding': forms.Textarea(attrs={'class': 'form-control', 'rows': 4}),
        }

class SectorApplicationForm(forms.ModelForm):
    class Meta:
        model = SectorApplication
        fields = ['sector', 'monthly_turnover', 'years_in_operation', 'employees_count']
        widgets = {
            'sector': forms.Select(attrs={'class': 'form-control'}),
            'monthly_turnover': forms.NumberInput(attrs={'class': 'form-control'}),
            'years_in_operation': forms.NumberInput(attrs={'class': 'form-control'}),
            'employees_count': forms.NumberInput(attrs={'class': 'form-control'}),
        }