Update app.py

app.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, HTTPException, Depends, Security
+from fastapi import FastAPI, HTTPException, Depends
 from fastapi.security import HTTPBearer
 from pydantic import BaseModel
 from transformers import GPT2LMHeadModel, GPT2TokenizerFast, GPT2Config
@@ -69,9 +69,15 @@ def classify_text(sentence: str):
 @app.post("/analyze")
 async def analyze_text(data: TextInput, token: str = Depends(bearer_scheme)):
     user_input = data.text.strip()
+
     if not user_input:
         raise HTTPException(status_code=400, detail="Text cannot be empty")
-
+    
+    # Check if there are at least two words
+    word_count = len(user_input.split())
+    if word_count < 2:
+        raise HTTPException(status_code=400, detail="Text must contain at least two words")
+    
     # The token is automatically extracted from the Authorization header
     # You can validate the token here if needed
     print(f"Received Bearer Token: {token}")