Update app.py
Browse files
app.py
CHANGED
@@ -24,8 +24,10 @@ english_generator = pipeline("text-generation", model="distilgpt2")
|
|
24 |
def detect_language(user_input):
|
25 |
try:
|
26 |
lang = detect(user_input)
|
|
|
27 |
return "hebrew" if lang == "he" else "english" if lang == "en" else "unsupported"
|
28 |
except:
|
|
|
29 |
return "unsupported"
|
30 |
|
31 |
|
@@ -58,13 +60,19 @@ async def root():
|
|
58 |
|
59 |
@app.post("/generate_response")
|
60 |
async def generate_text(request: Request):
|
61 |
-
|
62 |
-
|
63 |
-
|
64 |
-
|
65 |
-
|
66 |
-
|
67 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
68 |
|
69 |
|
70 |
# Run the server
|
|
|
24 |
def detect_language(user_input):
|
25 |
try:
|
26 |
lang = detect(user_input)
|
27 |
+
print(f"Detected language: {lang}")
|
28 |
return "hebrew" if lang == "he" else "english" if lang == "en" else "unsupported"
|
29 |
except:
|
30 |
+
print(f"Language detection error: {e}")
|
31 |
return "unsupported"
|
32 |
|
33 |
|
|
|
60 |
|
61 |
@app.post("/generate_response")
|
62 |
async def generate_text(request: Request):
|
63 |
+
try:
|
64 |
+
data = await request.json()
|
65 |
+
text = data.get("text", "").strip() # removes non relevant spaces
|
66 |
+
|
67 |
+
if not text:
|
68 |
+
return {"error": "No text provided"}
|
69 |
+
|
70 |
+
response = generate_response(text)
|
71 |
+
return {"response": response}
|
72 |
+
|
73 |
+
except Exception as e:
|
74 |
+
logging.error(f"Error processing request: {e}")
|
75 |
+
return {"error": "Invalid request. Please send JSON with a 'text' field."}
|
76 |
|
77 |
|
78 |
# Run the server
|