Update app.py

app.py

@@ -1,73 +1,11 @@
-from fastapi import FastAPI, HTTPException, Request
-from fastapi.responses import JSONResponse
-from pydantic import BaseModel
-from typing import Optional
-import logging
-from utils import *
+from flask import Flask, request, jsonify
+from transformers import pipeline
 
-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 = None
-# Define the request model
-class ArticleRequest(BaseModel):
-    titre_article: str
-    mot_cle_principal: str
-    ton_cible: str
-    optional_text : str = None
+app = Flask(__name__)
+classifier = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base", return_all_scores=True)
 
-# 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.
-     - optional_text: str - Optional text to include in 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")
-        
+@app.route('/classify', methods=['POST'])
+def classify():
+    text = request.json['text']
+    result = classifier(text)
+    return jsonify(result)