updated api

app/core/security.py

@@ -0,0 +1,7 @@
+from fastapi import Header, HTTPException
+
+API_KEY = "12345"
+
+def verify_api_key(x_api_key: str = Header(...)):
+    if x_api_key != API_KEY:
+        raise HTTPException(status_code=401, detail="Unauthorized")

app/main.py

@@ -1,75 +1,13 @@
-from fastapi import FastAPI, Body
-from pydantic import BaseModel
-from app import models, prompts
-import logging
-
-# Setup basic logging
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
+from fastapi import FastAPI
+from app.routers import analyze, paraphrase, translate, summarize
 
 app = FastAPI()
 
-class TextInput(BaseModel):
-    text: str
-    mode: str
-    tone: str = None
-    target_lang: str = None
-
 @app.get("/")
 def root():
-    return {"message": "Welcome to Build Your Own Grammarly API. Use /rewrite POST endpoint."}
-
-@app.post("/rewrite")
-def rewrite(input: TextInput):
-    text = input.text
-    mode = input.mode
-    logger.info(f"Received request | Mode: {mode} | Text: {text}")
-
-    try:
-        if mode == "grammar":
-            result = models.run_grammar_correction(text)
-            logger.info(f"Grammar corrected: {result}")
-            return {"result": result}
-
-        elif mode == "paraphrase":
-            prompt = prompts.paraphrase_prompt(text)
-            result = models.run_flan_prompt(prompt)
-            logger.info(f"Paraphrased: {result}")
-            return {"result": result}
-
-        elif mode == "clarity":
-            prompt = prompts.clarity_prompt(text)
-            result = models.run_flan_prompt(prompt)
-            logger.info(f"Clarity enhanced: {result}")
-            return {"result": result}
-
-        elif mode == "fluency":
-            prompt = prompts.fluency_prompt(text)
-            result = models.run_flan_prompt(prompt)
-            logger.info(f"Fluency improved: {result}")
-            return {"result": result}
-
-        elif mode == "tone" and input.tone:
-            prompt = prompts.tone_prompt(text, input.tone)
-            result = models.run_flan_prompt(prompt)
-            logger.info(f"Tone changed to {input.tone}: {result}")
-            return {"result": result}
-
-        elif mode == "translate" and input.target_lang:
-            result = models.run_translation(text, input.target_lang)
-            logger.info(f"Translated to {input.target_lang}: {result}")
-            return {"result": result}
-
-        elif mode == "pronoun":
-            prompt = prompts.pronoun_friendly_prompt(text)
-            result = models.run_flan_prompt(prompt)
-            logger.info(f"Inclusive pronoun version: {result}")
-            return {"result": result}
-
-        else:
-            logger.error("Invalid request parameters.")
-            return {"error": "Invalid request"}
+    return {"message": "Welcome to Grammafree API"}
 
-    except Exception as e:
-        logger.error(f"Error while processing: {str(e)}")
-        return {"error": str(e)}
+app.include_router(analyze.router)
+app.include_router(paraphrase.router)
+app.include_router(translate.router)
+app.include_router(summarize.router)

app/prompts.py

@@ -10,5 +10,14 @@ def fluency_prompt(text):
 def paraphrase_prompt(text):
     return f"Paraphrase: {text}"
 
+def summarize_prompt(text):
+    return f"Summarize: {text}"
+
 def pronoun_friendly_prompt(text):
     return f"Rewrite the text using inclusive and non-offensive pronouns: {text}"
+
+def active_voice_prompt(text):
+    return f"Detect if this is passive or active voice. If passive, suggest an active voice version: {text}"
+
+def tone_analysis_prompt(text):
+    return f"Analyze the tone of the following text and suggest improvements if needed: {text}"

app/routers/analyze.py

@@ -0,0 +1,21 @@
+from fastapi import APIRouter, Depends
+from pydantic import BaseModel
+from app import models, prompts
+from app.core.security import verify_api_key
+
+router = APIRouter()
+
+class AnalyzeInput(BaseModel):
+    text: str
+
+@router.post("/analyze", dependencies=[Depends(verify_api_key)])
+def analyze_text(input: AnalyzeInput):
+    text = input.text
+    return {
+        "grammar": models.run_grammar_correction(text),
+        "punctuation": models.run_flan_prompt(prompts.clarity_prompt(text)),
+        "sentence_correctness": models.run_flan_prompt(prompts.fluency_prompt(text)),
+        "tone_analysis": models.run_flan_prompt(prompts.tone_analysis_prompt(text)),
+        "voice": models.run_flan_prompt(prompts.active_voice_prompt(text)),
+        "inclusive_pronouns": models.run_flan_prompt(prompts.pronoun_friendly_prompt(text))
+    }

app/routers/paraphrase.py

@@ -0,0 +1,13 @@
+from fastapi import APIRouter, Depends
+from pydantic import BaseModel
+from app import models, prompts
+from app.core.security import verify_api_key
+
+router = APIRouter()
+
+class Input(BaseModel):
+    text: str
+
+@router.post("/paraphrase", dependencies=[Depends(verify_api_key)])
+def paraphrase(input: Input):
+    return {"result": models.run_flan_prompt(prompts.paraphrase_prompt(input.text))}

app/routers/summarize.py

@@ -0,0 +1,13 @@
+from fastapi import APIRouter, Depends
+from pydantic import BaseModel
+from app import models, prompts
+from app.core.security import verify_api_key
+
+router = APIRouter()
+
+class Input(BaseModel):
+    text: str
+
+@router.post("/summarize", dependencies=[Depends(verify_api_key)])
+def summarize(input: Input):
+    return {"result": models.run_flan_prompt(prompts.summarize_prompt(input.text))}

app/routers/translate.py

@@ -0,0 +1,14 @@
+from fastapi import APIRouter, Depends
+from pydantic import BaseModel
+from app import models
+from app.core.security import verify_api_key
+
+router = APIRouter()
+
+class TranslateInput(BaseModel):
+    text: str
+    target_lang: str
+
+@router.post("/translate", dependencies=[Depends(verify_api_key)])
+def translate(input: TranslateInput):
+    return {"result": models.run_translation(input.text, input.target_lang)}