Update main.py

main.py

@@ -109,11 +109,11 @@ def root():
 class ReviewInput(BaseModel):
     text: str
     model: str = "distilbert-base-uncased-finetuned-sst-2-english"
-    industry: str = "Generic"
+    industry: Optional[str] = None
     aspects: bool = False
-    follow_up: str = None
-    product_category: str = None
-    device: str = None
+    follow_up: Optional[str] = None
+    product_category: Optional[str] = None
+    device: Optional[str] = None
 
 class BulkReviewInput(BaseModel):
     reviews: list[str]
@@ -141,6 +141,11 @@ sentiment_pipelines = {
     "nlptown/bert-base-multilingual-uncased-sentiment": pipeline("sentiment-analysis", model="nlptown/bert-base-multilingual-uncased-sentiment")
 }
 
+def auto_fill(value: Optional[str], default: str = "Generic") -> str:
+    if not value or value.strip().lower() == "auto-detect":
+        return default
+    return value
+
 @app.post("/bulk/")
 async def bulk(data: BulkReviewInput, x_api_key: str = Header(None)):
     if x_api_key != VALID_API_KEY:
@@ -164,22 +169,14 @@ async def bulk(data: BulkReviewInput, x_api_key: str = Header(None)):
             "sentiment": label,
             "emotion": emotions[i][0]["label"],
             "aspects": [],
-            "product_category": data.product_category[i] if data.product_category else None,
-            "device": data.device[i] if data.device else None,
-            "industry": data.industry[i] if data.industry else None,
+            "product_category": auto_fill(data.product_category[i]) if data.product_category else None,
+            "device": auto_fill(data.device[i], "Web") if data.device else None,
+            "industry": auto_fill(data.industry[i]) if data.industry else None,
         }
         results.append(result)
 
     return {"results": results}
 
-@app.post("/upload/")
-async def upload(file: UploadFile = File(...)):
-    content = await file.read()
-    decoded = content.decode("utf-8")
-    reader = csv.DictReader(StringIO(decoded))
-    reviews = [row["review"] for row in reader if "review" in row]
-    return {"count": len(reviews), "sample": reviews[:3]}
-
 @app.post("/analyze/")
 async def analyze(request: Request, data: ReviewInput, x_api_key: str = Header(None), download: str = None):
     if x_api_key != VALID_API_KEY:
@@ -214,9 +211,9 @@ async def analyze(request: Request, data: ReviewInput, x_api_key: str = Header(N
         "emotion": emotion,
         "aspects": aspects_list,
         "follow_up": follow_up_response,
-        "product_category": data.product_category,
-        "device": data.device,
-        "industry": data.industry
+        "product_category": auto_fill(data.product_category),
+        "device": auto_fill(data.device, "Web"),
+        "industry": auto_fill(data.industry)
     }
 
 @app.post("/translate/")