Skip to content
7 changes: 6 additions & 1 deletion webapp/dface.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,4 +27,9 @@ def dface(path):

else:
print('someting error')
print(path)
print(path)

def gallery(path):
img = cv2.imread(path, 1)
cv2.imwrite(path, img)

7 changes: 1 addition & 6 deletions webapp/forms.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,7 @@
from django import forms
from .models import ImageUploadModel

class UploadImageForm(forms.Form):
title=forms.CharField(max_length=50)
#file = forms.FileField()
image=forms.ImageField()

class ImageUploadForm(forms.ModelForm):
class Meta:
model = ImageUploadModel
fields = ('description', 'document', )
fields = ('title', 'description', 'document', )
3 changes: 2 additions & 1 deletion webapp/models.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
from django.db import models

class ImageUploadModel(models.Model):
title = models.CharField(max_length=50, blank=True)
description = models.CharField(max_length=255, blank=True)
document = models.ImageField(upload_to='documents/')
uploaded_at = models.DateTimeField(auto_now_add=True)
uploaded_at = models.DateTimeField(auto_now_add=True)
19 changes: 19 additions & 0 deletions webapp/templates/gallery.html
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
{% extends 'base.html' %}
{% load webapp_tags %}

{% block content %}

<div width: 400px; height: 200px; color: #white;>
<!-- image upload start-->
{% if documents %}
<a href="{{ MEDIA_URL }}documents/{{ img_list }}">
<img src="{{ MEDIA_URL }}documents/{{ img_list }}">
</a>

{% endif %}

<p><a href="{% url 'home' %}">Return to home</a></p>
<!-- image upload end -->
</div>

{% endblock %}
1 change: 1 addition & 0 deletions webapp/templatetags/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@

11 changes: 11 additions & 0 deletions webapp/templatetags/webapp_tags
Original file line number Diff line number Diff line change
@@ -0,0 +1,11 @@
from django import template

register = template.Library()

@register.filter(name='split')
def split(str, key):
return str.split(key)

@register.filter
def get_by_index(a, i):
return a[i]
4 changes: 2 additions & 2 deletions webapp/urls.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,8 +7,8 @@
urlpatterns = [
url('^$', views.home, name='home'),
url('detection/', views.detection, name='detection'),
url('upload/', views.upload, name='upload'),
url('gallery/', views.gallery, name='gallery'),
url('about/', views.about, name='about'),
]

urlpatterns += static(settings.MEDIA_URL, document_root=settings.MEDIA_ROOT)
urlpatterns += static(settings.MEDIA_URL, document_root=settings.MEDIA_ROOT)
78 changes: 39 additions & 39 deletions webapp/views.py
Original file line number Diff line number Diff line change
@@ -1,39 +1,39 @@
from django.shortcuts import render, redirect
from django.conf import settings
from django.core.files.storage import FileSystemStorage
from .forms import UploadImageForm
from .forms import ImageUploadForm
from .dface import dface

def home(request):
return render(request, 'home.html')

def about(request):
return render(request, 'about.html')

def upload(request):
form = UploadImageForm(request.POST, request.FILES)
if form.is_valid():
myfile = request.FILES['image']
fs = FileSystemStorage()
filename = fs.save(myfile.name, myfile)
uploaded_file_url = fs.url(filename)
return render(request, 'upload.html', {'form': form, 'uploaded_file_url': uploaded_file_url})
else:
form = UploadImageForm()
return render(request, 'upload.html', {'form': form})

def detection(request):
form = ImageUploadForm(request.POST, request.FILES)
if request.method == 'POST':
if form.is_valid():
post = form.save(commit=False)
post.save()
imageURL = settings.MEDIA_URL + form.instance.document.name
dface(settings.MEDIA_ROOT_URL + imageURL)

return render(request, 'detection.html', {'form':form, 'post':post})
else:
form = ImageUploadForm()
return render(request, 'detection.html',{'form':form})
import os
from django.shortcuts import render, redirect
from django.conf import settings
from django.core.files.storage import FileSystemStorage
from .forms import ImageUploadForm
#from .forms import GalleryForm
from .dface import dface
#from .dface import gallery
from cv2 import cv2
def home(request):
return render(request, 'home.html')
def about(request):
return render(request, 'about.html')
def detection(request):
form = ImageUploadForm(request.POST, request.FILES)
if request.method == 'POST':
if form.is_valid():
post = form.save(commit=False)
post.save()
imageURL = settings.MEDIA_URL + form.instance.document.name
dface(settings.MEDIA_ROOT_URL + imageURL)
return render(request, 'detection.html', {'form':form, 'post':post})
else:
form = ImageUploadForm()
return render(request, 'detection.html',{'form':form})
def gallery(request):
imageURL = settings.MEDIA_ROOT_URL + settings.MEDIA_URL + 'documents/'
img_list = os.listdir(imageURL)
for file in img_list:
img = cv2.imread(imageURL+file, 1)
# cv2.imwrite(imageURL+file, img)
return render(request, "gallery.html", {'documents': img_list})