Update app.py

app.py

@@ -1,68 +1,71 @@
-from fastapi import FastAPI, HTTPException, Request
-from fastapi.responses import JSONResponse
-from pydantic import BaseModel
-from typing import Optional
-import logging
-from utils import *
-
-app = FastAPI()
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
-# Define the request model
-class ArticleRequesteng(BaseModel):
-    article_title: str
-    main_keyword: str
-    target_tone: str
-# Define the request model
-class ArticleRequest(BaseModel):
-    titre_article: str
-    mot_cle_principal: str
-    ton_cible: str
-
-# Define the response model
-class ArticleResponse(BaseModel):
-    article: str
-
-@app.post("/generate_article_fr", response_model=ArticleResponse)
-async def generate_article(request: ArticleRequest):
-    """
-    Endpoint to generate a French SEO article.
-    Parameters:
-    - titre_article: str - The title of the article.
-    - mot_cle_principal: str - The main keyword for the article.
-    - ton_cible: str - The target tone of the article.
-    """
-    try:
-        article = create_pipeline_fr(request.titre_article, request.mot_cle_principal, request.ton_cible)
-        return ArticleResponse(article=article)
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
-
-@app.post("/generate_article_eng", response_model=ArticleResponse)
-async def generate_article_eng(request: ArticleRequesteng):
-    """
-    Endpoint to generate an SEO article.
-    Parameters:
-    - article_title: str - The title of the article.
-    - main_keyword: str - The main keyword for the article.
-    - target_tone: str - The target tone of the article.
-    """
-    try:
-        # Basic validation of the input
-        if not request.article_title or not request.main_keyword:
-            raise HTTPException(status_code=400, detail="Title and main keyword are required")
-
-        article = create_pipeline(request.article_title, request.main_keyword, request.target_tone)
-        
-        # Ensure the response is not empty
-        if not article:
-            raise HTTPException(status_code=204, detail="Generated article is empty")
-
-        return ArticleResponse(article=article)
-    
-    except HTTPException as http_exc:
-        logger.error(f"HTTP Exception: {http_exc.detail}")
-        raise http_exc
-    except Exception as e:
-        logger.error(f"Unhandled Exception: {str(e)}")
+from fastapi import FastAPI, HTTPException, Request
+from fastapi.responses import JSONResponse
+from pydantic import BaseModel
+from typing import Optional
+import logging
+from utils import *
+
+app = FastAPI()
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+# Define the request model
+class ArticleRequesteng(BaseModel):
+    article_title: str
+    main_keyword: str
+    target_tone: str
+    optional_text: str
+# Define the request model
+class ArticleRequest(BaseModel):
+    titre_article: str
+    mot_cle_principal: str
+    ton_cible: str
+    optional_text : str
+
+# Define the response model
+class ArticleResponse(BaseModel):
+    article: str
+
+@app.post("/generate_article_fr", response_model=ArticleResponse)
+async def generate_article(request: ArticleRequest):
+    """
+    Endpoint to generate a French SEO article.
+    Parameters:
+    - titre_article: str - The title of the article.
+    - mot_cle_principal: str - The main keyword for the article.
+    - ton_cible: str - The target tone of the article.
+    - optional_text: str - Optional text to include in the article.
+    """
+    try:
+        article = create_pipeline_fr(request.titre_article, request.mot_cle_principal, request.ton_cible,request.optional_text)
+        return ArticleResponse(article=article)
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.post("/generate_article_eng", response_model=ArticleResponse)
+async def generate_article_eng(request: ArticleRequesteng):
+    """
+    Endpoint to generate an SEO article.
+    Parameters:
+    - article_title: str - The title of the article.
+    - main_keyword: str - The main keyword for the article.
+    - target_tone: str - The target tone of the article.
+    """
+    try:
+        # Basic validation of the input
+        if not request.article_title or not request.main_keyword:
+            raise HTTPException(status_code=400, detail="Title and main keyword are required")
+
+        article = create_pipeline(request.article_title, request.main_keyword, request.target_tone,request.optional_text)
+        
+        # Ensure the response is not empty
+        if not article:
+            raise HTTPException(status_code=204, detail="Generated article is empty")
+
+        return ArticleResponse(article=article)
+    
+    except HTTPException as http_exc:
+        logger.error(f"HTTP Exception: {http_exc.detail}")
+        raise http_exc
+    except Exception as e:
+        logger.error(f"Unhandled Exception: {str(e)}")
         raise HTTPException(status_code=500, detail="An internal server error occurred")