DeMaking commited on
Commit
49a5c30
·
verified ·
1 Parent(s): 8ad9118

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -7
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
- data = await request.json()
62
- text = data.get("text", "")
63
- if not text:
64
- return {"error": "No text provided"}
65
-
66
- response = generate_response(text)
67
- return {"response": response}
 
 
 
 
 
 
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