Updated

GBU31 · Dec 4, 2023 · 3065bce · 3065bce
1 parent 105a475
commit 3065bce
Show file tree

Hide file tree

Showing 9 changed files with 17 additions and 13 deletions.
diff --git a/images/test.jpg → images/test.png b/images/test.jpg → images/test.png
diff --git a/output/output.avi → images_two/test.png b/output/output.avi → images_two/test.png
diff --git a/janus_ai/settings.py b/janus_ai/settings.py
@@ -24,7 +24,7 @@
 SECRET_KEY = 'django-insecure-766bec8h)4wk49djosnvis1=fyb0y=2qlef7(@et_-fq#72@to'
 
 # SECURITY WARNING: don't run with debug turned on in production!
-DEBUG = False
+DEBUG = True
 
 ALLOWED_HOSTS = ['*']
 

diff --git a/main/DeepFake/main.py b/main/DeepFake/main.py
@@ -146,11 +146,12 @@ def set_src_image(image):
     indexes_triangles = get_triangles(convexhull=src_convexHull,
                                                   landmarks_points=src_landmark_points,
                                                   np_points=src_np_points)
-
+i = 0
 try:
     set_src_image(image)
 
-    while True:
+    while i < 2:
+        i += 1
         global src_image, src_image_gray, src_mask, src_landmark_points, src_np_points, src_convexHull, indexes_triangles
 
         _, dest_image = cap.read()

diff --git a/main/migrations/0001_initial.py b/main/migrations/0001_initial.py
@@ -1,4 +1,4 @@
-# Generated by Django 4.2 on 2023-04-14 09:57
+# Generated by Django 4.0.3 on 2023-12-04 02:03
 
 from django.db import migrations, models
 
@@ -15,8 +15,8 @@ class Migration(migrations.Migration):
             name='FileModel',
             fields=[
                 ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
-                ('image', models.ImageField(upload_to='images/')),
-                ('video', models.FileField(upload_to='videos/')),
+                ('image_one', models.ImageField(upload_to='images/')),
+                ('image_two', models.ImageField(upload_to='images_two/')),
             ],
         ),
     ]
diff --git a/main/models.py b/main/models.py
@@ -1,5 +1,5 @@
 from django.db import models
 
 class FileModel(models.Model):
-    image = models.ImageField(upload_to='images/')
-    video = models.FileField(upload_to='videos/')
+    image_one = models.ImageField(upload_to='images/')
+    image_two = models.ImageField(upload_to='images_two/')
diff --git a/main/serializers.py b/main/serializers.py
@@ -4,4 +4,4 @@
 class FileSerializer(serializers.ModelSerializer):
     class Meta:
         model = FileModel
-        fields = ['image', 'video']
+        fields = '__all__'
diff --git a/main/views.py b/main/views.py
@@ -28,9 +28,12 @@ def upload_file(self, request):
         serializer = FileSerializer(data=request.data)
         if serializer.is_valid():
             serializer.save()
-            os.system(f'python3 main/DeepFake/main.py videos/{str(request.data["video"])} images/{str(request.data["image"])} output/{random_string}.avi')
-            response = FileResponse(open('output/'+random_string + '.avi', 'rb'))
-            response['Content-Disposition'] = f'attachment; filename=output/"{random_string}.avi"'
-            os.system(f'rm -rf videos/{str(request.data["video"])} images/{str(request.data["image"])} output/{random_string}.avi')
+
+            os.system(f'python3 main/DeepFake/main.py images_two/{str(request.data["image_two"])} images/{str(request.data["image_one"])} output/{random_string}.png')
+            response = FileResponse(open('output/'+random_string + '.png', 'rb'))
+            response['Content-Disposition'] = f'attachment; filename=output/"{random_string}.png"'
+            os.system(f'rm -rf images_two/{str(request.data["image_two"])} images/{str(request.data["image_one"])} output/{random_string}.png')
             return response
+
+                # return Response(serializer.errors, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
         return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
diff --git a/videos/test.mp4 → output/test.png b/videos/test.mp4 → output/test.png