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
|