From 18fc472d48631562f53b453a75ebde78f5628f9b Mon Sep 17 00:00:00 2001 From: Vinay Jain <48233751+vinayjain18@users.noreply.github.com> Date: Thu, 1 Aug 2024 02:10:43 +0530 Subject: [PATCH] Updated --- creator/views.py | 8 +++++++- templates/dashboard.html | 13 +++++++++++++ templates/history.html | 18 ++++++++++++++++-- 3 files changed, 36 insertions(+), 3 deletions(-) diff --git a/creator/views.py b/creator/views.py index 68a7e1c..ce34003 100644 --- a/creator/views.py +++ b/creator/views.py @@ -36,7 +36,7 @@ def dashboard(request): response = generate_response(ig_username, industry, niche, about_ig_page, weeks) if response.startswith("```html") and response.endswith("```"): response = response[7:-3] # Remove the leading and trailing markdown code blocks - messages.success(request, "Content plan generated successfully") + messages.success(request, "Content plan generated successfully. Scroll down to bottom to view the plan.") input_context = f"IG Username: {ig_username}\nIndustry: {industry}\nNiche: {niche}\nAbout IG Page: {about_ig_page}\nWeeks: {weeks}" ContentGeneration.objects.create(email=request.user.email, input_context=input_context, content_output=response).save() return render(request, 'dashboard.html', {'response': response}) @@ -51,6 +51,12 @@ def history(request): @login_required(login_url='/signin/') def download_content(request, content_id): content_generation = get_object_or_404(ContentGeneration, pk=content_id) + + # Check if the logged-in user is the owner of the content + if content_generation.email != request.user.email: + messages.error(request, "You do not have permission to download this content.") + return redirect('/dashboard/') # Redirect to dashboard or appropriate page + content_html = content_generation.content_output # Use BeautifulSoup to strip HTML tags diff --git a/templates/dashboard.html b/templates/dashboard.html index a4b0bee..bbffc21 100644 --- a/templates/dashboard.html +++ b/templates/dashboard.html @@ -98,6 +98,19 @@
+ + {% if messages %} +
+
+ {% for message in messages %} +
+ {{ message }} +
+ {% endfor %} +
+
+ {% endif %} +
diff --git a/templates/history.html b/templates/history.html index fe09e01..70733d3 100644 --- a/templates/history.html +++ b/templates/history.html @@ -41,9 +41,9 @@