Update main.py

main.py

@@ -6,7 +6,7 @@ from io import BytesIO
 
 app = FastAPI()
 
-# গুগল-এর একটি ফাস্ট এবং পাওয়ারফুল ভিশন এআই মডেল
+# গুগল-এর ভিশন এআই মডেল
 classifier = pipeline("image-classification", model="google/vit-base-patch16-224")
 
 @app.get("/")
@@ -16,31 +16,20 @@ def read_root():
 @app.get("/analyze")
 def analyze(url: str):
     try:
-        # টিকটকের থাম্বনেইল সরাসরি ডাউনলোড করে স্ক্যান করা
-        res = requests.get(url, timeout=10)
+        # ব্রাউজার সেজে রিকোয়েস্ট পাঠানো যেন টিকটক ব্লক না করে
+        headers = {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64 AppleWebKit/537.36)"}
+        res = requests.get(url, headers=headers, timeout=15)
         img = Image.open(BytesIO(res.content)).convert("RGB")
         
         # AI Processing
         results = classifier(img)
-        
-        # Confidence Check (85% = 0.85)
         top_score = results[0]['score']
         
         if top_score >= 0.85:
-            # যদি এআই প্রায় ১০০% নিশ্চিত হয়, তবে শুধু ১টি নাম দেবে
-            return {
-                "success": True, 
-                "confidence": f"High ({int(top_score*100)}%)", 
-                "names": [results[0]['label'].title()]
-            }
+            return {"success": True, "confidence": f"High ({int(top_score*100)}%)", "names": [results[0]['label'].title()]}
         else:
-            # যদি এআই পুরোপুরি নিশ্চিত না হয়, তবে সম্ভাব্য সেরা ৩টি নাম দেবে
             top_3 = [r['label'].title() for r in results[:3]]
-            return {
-                "success": True, 
-                "confidence": "Low", 
-                "names": top_3
-            }
+            return {"success": True, "confidence": "Low", "names": top_3}
             
     except Exception as e:
         return {"success": False, "error": str(e)}