Update app.py

app.py

@@ -1,9 +1,11 @@
 import os
+import logging
 from flask import Flask, request, jsonify
 from transformers import pipeline
 from langdetect import detect
 from huggingface_hub import login
 
+# Initialize Flask app
 app = Flask(__name__)
 
 # Gets the Token from secrect
@@ -12,28 +14,63 @@ hf_hub_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
 # Logging in
 login(token=hf_hub_token)
 
-# Load models
-hebrew_model = pipeline("text-generation", model="onlplab/alephbert-base", is_decoder=True)
-english_model = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.3")
+# Load Hebrew models (classification + text generation)
+hebrew_classifier = pipeline("text-classification", model="onlplab/alephbert-base")
+hebrew_generator = pipeline("text2text-generation", model="google/mt5-small")
 
+# Load English models
+english_classifier = pipeline("text-classification", model="mistralai/Mistral-7B-Instruct-v0.3")
+english_generator = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.3")
+
+# Function to detect language
+def detect_language(user_input):
+    try:
+        lang = detect(user_input)
+        if lang == "he":
+            return "hebrew"
+        elif lang == "en":
+            return "english"
+        else:
+            return "unsupported"
+    except:
+        return "unsupported"
+
+
+
+# Function to generate response based on language
+def generate_response(text):
+    language = detect_language(text)
+
+    if language == "hebrew":
+        classification = hebrew_classifier(text)
+        response = hebrew_generator(text, max_length=100)[0]["generated_text"]
+    elif language == "english":
+        classification = english_classifier(text)
+        response = english_generator(text, max_length=100)[0]["generated_text"]
+    else:
+        response = "❌ Sorry, I only support Hebrew and English."
+
+    return response
+
+
+# Flask endpoint for processing text input
 @app.route("/ask", methods=["POST"])
 def ask():
     data = request.json
     user_input = data.get("text", "")
 
-    # Detect language 
-    language = detect(user_input)
+    if not user_input:
+        return jsonify({"error": "No text provided"}), 400
 
-    if language == 'he':
-        model = hebrew_model
-    elif language == 'en':
-        model = english_model
-    else:
-        print("Decision Making Helper BOT currently supports Hebrew and English Languages")
+    response = generate_response(user_input)
+    return jsonify({"response": response})
 
-    # Create an answer from the model
-    response = model(user_input, max_length=100, do_sample=True)
-    return jsonify({"response": response[0]['generated_text']})
+    
+# Root endpoint
+@app.route("/")
+def home():
+    return "Decision Helper Bot API is running!"
 
 if __name__ == "__main__":
+    logging.basicConfig(level=logging.INFO)
     app.run(host="0.0.0.0", port=7860)