Update app.py

app.py

@@ -36,12 +36,29 @@ def detect_language(user_input):
 
 
 # Function to generate a response
+# def generate_response(text):
+#     language = detect_language(text)
+#     if language == "hebrew":
+#         return hebrew_generator(text, max_length=100, truncation=True)[0]["generated_text"]
+#     elif language == "english":
+#         return english_generator(text, max_length=100, truncation=True)[0]["generated_text"]
+#     return "Sorry, I only support Hebrew and English."
+
+
 def generate_response(text):
     language = detect_language(text)
+    print(f"Detected language: {language}")  # Debugging
+
     if language == "hebrew":
-        return hebrew_generator(text, max_length=100, truncation=True)[0]["generated_text"]
+        output = hebrew_generator(text, max_length=100, truncation=True)
+        print(f"Hebrew model output: {output}")  # Debugging
+        return output[0]["generated_text"]
+    
     elif language == "english":
-        return english_generator(text, max_length=100, truncation=True)[0]["generated_text"]
+        output = english_generator(text, max_length=100, truncation=True)
+        print(f"English model output: {output}")  # Debugging
+        return output[0]["generated_text"]
+    
     return "Sorry, I only support Hebrew and English."
 
 
@@ -62,22 +79,41 @@ async def root():
     return {"message": "Decision Helper API is running!"}
 
 
+# @app.post("/generate_response")
+# async def generate_text(request: Request):
+#     try:
+#         data = await request.json()
+#         text = data.get("text", "").strip()  # removes non-relevant spaces
+
+#         if not text:
+#             return {"error": "No text provided"}
+
+#         response = generate_response(text)
+#         return {"response": response}
+
+#     except Exception as e:
+#         logging.error(f"Error processing request: {e}")
+#         return {"error": "Invalid request. Please send JSON with a 'text' field."}
+
+
 @app.post("/generate_response")
 async def generate_text(request: Request):
     try:
         data = await request.json()
-        text = data.get("text", "").strip()  # removes non-relevant spaces
+        logging.info(f"Received request: {data}")  # Log the request data
 
+        text = data.get("text", "").strip()   # removes non-relevant spaces
         if not text:
             return {"error": "No text provided"}
 
         response = generate_response(text)
+        logging.info(f"Generated response: {response}")  # Log the response
         return {"response": response}
 
     except Exception as e:
         logging.error(f"Error processing request: {e}")
         return {"error": "Invalid request. Please send JSON with a 'text' field."}
-
+        
 
 # Run the server
 if __name__ == "__main__":