Add inspiration tab and integrate into generation

core/admin.py

@@ -6,6 +6,7 @@
     BlogPostTitleSuggestion,
     EmailSent,
     GeneratedBlogPost,
+    Inspiration,
     Profile,
     Project,
     ProjectPage,
@@ -80,3 +81,4 @@ class ReferrerBannerAdmin(admin.ModelAdmin):
 admin.site.register(AutoSubmissionSetting)
 admin.site.register(ProjectPage)
 admin.site.register(EmailSent)
+admin.site.register(Inspiration)

core/agent_system_prompts.py

@@ -153,6 +153,41 @@ def add_target_keywords(ctx: RunContext[BlogPostGenerationContext]) -> str:
         return ""
 
 
+def add_inspirations(ctx: RunContext[BlogPostGenerationContext]) -> str:
+    if not ctx.deps.inspirations:
+        return ""
+
+    inspiration_sections = []
+    inspiration_sections.append(
+        """
+        INSPIRATION SOURCES:
+        The user has shared the following blog posts and blogs as inspiration sources
+        they would like to emulate in style, tone, and structure.
+        Use these as reference points when generating content:
+        """
+    )
+
+    for inspiration in ctx.deps.inspirations:
+        inspiration_sections.append(
+            f"""
+            - Title: {inspiration.title}
+              URL: {inspiration.url}
+              Description: {inspiration.description or 'No description provided'}
+            """
+        )
+
+    inspiration_sections.append(
+        """
+        IMPORTANT: Draw inspiration from these sources' writing style, tone, structure,
+        and approach while creating content that is unique and appropriate for the current project.
+        Consider how these inspirations structure their content, engage their audience,
+        and present information.
+        """
+    )
+
+    return "\n".join(inspiration_sections)
+
+
 def add_webpage_content(ctx: RunContext[WebPageContent]) -> str:
     return (
         "Web page content:"

core/agents.py

@@ -4,6 +4,7 @@
 from pydantic_ai.providers.openai import OpenAIProvider
 
 from core.agent_system_prompts import (
+    add_inspirations,
     add_language_specification,
     add_project_details,
     add_project_pages,
@@ -51,6 +52,7 @@ def only_return_the_edited_content() -> str:
 content_editor_agent.system_prompt(add_title_details)
 content_editor_agent.system_prompt(add_language_specification)
 content_editor_agent.system_prompt(add_target_keywords)
+content_editor_agent.system_prompt(add_inspirations)
 
 
 ########################################################

core/api/schemas.py

@@ -262,3 +262,25 @@ class ToggleProjectPageAlwaysUseOut(Schema):
     status: str
     always_use: bool
     message: str | None = None
+
+
+class AddInspirationIn(Schema):
+    project_id: int
+    url: str
+
+
+class AddInspirationOut(Schema):
+    status: str
+    message: str | None = None
+    inspiration_id: int | None = None
+    title: str | None = None
+    url: str | None = None
+
+
+class DeleteInspirationIn(Schema):
+    inspiration_id: int
+
+
+class DeleteInspirationOut(Schema):
+    status: str
+    message: str | None = None

core/api/views.py

@@ -9,12 +9,16 @@
 from core.api.auth import session_auth, superuser_api_auth
 from core.api.schemas import (
     AddCompetitorIn,
+    AddInspirationIn,
+    AddInspirationOut,
     AddKeywordIn,
     AddKeywordOut,
     AddPricingPageIn,
     BlogPostIn,
     BlogPostOut,
     CompetitorAnalysisOut,
+    DeleteInspirationIn,
+    DeleteInspirationOut,
     DeleteProjectKeywordIn,
     DeleteProjectKeywordOut,
     FixGeneratedBlogPostIn,
@@ -52,6 +56,7 @@
     Competitor,
     Feedback,
     GeneratedBlogPost,
+    Inspiration,
     Keyword,
     Project,
     ProjectKeyword,
@@ -1056,3 +1061,84 @@ def toggle_project_page_always_use(request: HttpRequest, data: ToggleProjectPage
             "always_use": False,
             "message": f"Failed to toggle always use: {str(error)}",
         }
+
+
+@api.post("/inspirations/add", response=AddInspirationOut, auth=[session_auth])
+def add_inspiration(request: HttpRequest, data: AddInspirationIn):
+    """Add a new inspiration link to a project."""
+    profile = request.auth
+    project = get_object_or_404(Project, id=data.project_id, profile=profile)
+
+    url_to_add = data.url.strip()
+
+    if not url_to_add:
+        return {"status": "error", "message": "URL cannot be empty"}
+
+    if not url_to_add.startswith(("http://", "https://")):
+        return {"status": "error", "message": "URL must start with http:// or https://"}
+
+    try:
+        if Inspiration.objects.filter(project=project, url=url_to_add).exists():
+            return {"status": "error", "message": "This inspiration already exists for your project"}
+
+        inspiration = Inspiration.objects.create(project=project, url=url_to_add)
+
+        inspiration.get_page_content()
+
+        logger.info(
+            "[Add Inspiration] Successfully added inspiration",
+            inspiration_id=inspiration.id,
+            project_id=project.id,
+            url=url_to_add,
+        )
+
+        return {
+            "status": "success",
+            "inspiration_id": inspiration.id,
+            "title": inspiration.title,
+            "url": inspiration.url,
+            "message": "Inspiration added successfully!",
+        }
+
+    except Exception as e:
+        logger.error(
+            "Failed to add inspiration",
+            error=str(e),
+            exc_info=True,
+            project_id=project.id,
+            url=url_to_add,
+        )
+        return {"status": "error", "message": f"An unexpected error occurred: {str(e)}"}
+
+
+@api.post("/inspirations/delete", response=DeleteInspirationOut, auth=[session_auth])
+def delete_inspiration(request: HttpRequest, data: DeleteInspirationIn):
+    """Delete an inspiration from a project."""
+    profile = request.auth
+
+    try:
+        inspiration = get_object_or_404(
+            Inspiration, id=data.inspiration_id, project__profile=profile
+        )
+
+        inspiration_url = inspiration.url
+        inspiration.delete()
+
+        logger.info(
+            "[Delete Inspiration] Successfully deleted inspiration",
+            inspiration_id=data.inspiration_id,
+            url=inspiration_url,
+            profile_id=profile.id,
+        )
+
+        return {"status": "success", "message": f"Inspiration deleted successfully"}
+
+    except Exception as e:
+        logger.error(
+            "Failed to delete inspiration",
+            error=str(e),
+            exc_info=True,
+            inspiration_id=data.inspiration_id,
+            profile_id=profile.id,
+        )
+        return {"status": "error", "message": f"Failed to delete inspiration: {str(e)}"}