Update app.py

app.py

@@ -37,14 +37,16 @@ def generate_response(text):
     language = detect_language(text)
     print(f"Detected language: {language}, ", f"current time: {current_time_gmt()}")
      
-    if language == "hebrew" or language == "english":
+    if language == "hebrew":  # or language == "english":
+        content = "תענה בקצרה אבל תשתף את תהליך קבלת ההחלטות שלך, " + text
+        print("content: ", content)
         messages = [
-            { "role": "user", "content": text }
+            { "role": "user", "content": content }
         ]
         print(f"Messages: {messages}, ", f"current time: {current_time_gmt()}")
         
         completion = client.chat.completions.create(    
-            model="mistralai/Mistral-Nemo-Instruct-2407", 
+            model="onlplab/alephbert-base", 
             # model="microsoft/Phi-3.5-mini-instruct", # good in english. not so good in hebrew
         	messages=messages, 
         	max_tokens=2048,
@@ -53,6 +55,22 @@ def generate_response(text):
         )
         #print("\ncompletion: ", completion.choices[0].message.content, f"\ncurrent time: {current_time_gmt()}")
         return completion.choices[0].message.content
+    elif language ==  "english":
+        content = "keep it short but tell your decision making process, " + text
+        print("content: ", content)
+        messages = [
+            { "role": "user", "content": content }
+        ]
+        print(f"Messages: {messages}, ", f"current time: {current_time_gmt()}")
+        
+        completion = client.chat.completions.create(    
+            model="mistralai/Mistral-Nemo-Instruct-2407", 
+        	messages=messages, 
+        	max_tokens=2048,
+            temperature=0.5,
+            top_p=0.7 
+        )
+        return completion.choices[0].message.content
         
     return "Sorry, I only support Hebrew and English."