Update app.py

app.py

@@ -3,19 +3,21 @@ import os
 from fastapi import FastAPI, Request
 from contextlib import asynccontextmanager
 from transformers import pipeline
-from langdetect import detect  # to change to langid
 import langid
 from huggingface_hub import login
 import socket
 
+
 # Global variables
 HF_HUB_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")
 
+
 # Verify Hugging Face token
 if not HF_HUB_TOKEN:
      raise ValueError("Missing Hugging Face API token. Please set HUGGINGFACEHUB_API_TOKEN in environment variables.")
 login(token=HF_HUB_TOKEN)
 
+
 # Load Hebrew and English text generation models
 hebrew_generator = pipeline("text-generation", model="Norod78/hebrew-gpt_neo-small")
 english_generator = pipeline("text-generation", model="distilgpt2")
@@ -25,10 +27,10 @@ english_generator = pipeline("text-generation", model="distilgpt2")
 def detect_language(user_input):
     try:
        # lang = detect(user_input)
-        lang = langid.classify(user_input)
+        langת _ = langid.classify(user_input) # langid.classify returns a tuple (language, confidence)
         print(f"Detected language: {lang}") 
         return "hebrew" if lang == "he" else "english" if lang == "en" else "unsupported"
-    except:
+    except Exception as e:
         print(f"Language detection error: {e}")
         return "unsupported"
 
@@ -64,7 +66,7 @@ async def root():
 async def generate_text(request: Request):
     try:
         data = await request.json()
-        text = data.get("text", "").strip()  # removes non relevant spaces
+        text = data.get("text", "").strip()  # removes non-relevant spaces
 
         if not text:
             return {"error": "No text provided"}