Update app.py

app.py

@@ -114,18 +114,9 @@ def query(payload):
 	response = requests.post(API_URL, headers=headers, json=payload)
 	return response.json()
 
-# def conversation_predict(text):
-#     return interface([text])[0]
-
-def ai_predicted(user_input):
-    if user_input.lower() == 'exit':
-        return "Goodbye!"
-
-    user_input = translate_Singlish_to_sinhala(user_input)
-    user_input = transliterate_to_sinhala(user_input)
-    user_input = translate_sinhala_to_english(user_input)
+def conversation_predict(text):
     response_json = query({
-    	"inputs": f"{user_input}",
+    	"inputs": f"{text}",
     })
     if 'generated_text' in response_json[0]:
         ai_response = response_json[0]['generated_text']
@@ -133,11 +124,30 @@ def ai_predicted(user_input):
         ai_response = response_json[0]['text']
     else:
         ai_response = response_json[0]
-    # ai_response = conversation_predict(user_input)
+
+    return ai_response
+
+def ai_predicted(user_input):
+    if user_input.lower() == 'exit':
+        print("Goodbye!")
+        break
+
+    user_input = translate_Singlish_to_sinhala(user_input)
+    user_input = transliterate_to_sinhala(user_input)
+    user_input = translate_sinhala_to_english(user_input)
+    print("You(english):", user_input,"\n")
+
+    # Get AI response
+    ai_response = conversation_predict(user_input)
+
+    # Split the AI response into separate lines
     # ai_response_lines = ai_response.split("</s>")
+    print("AI(English):", ai_response,"\n")
 
     response = translate_english_to_sinhala(ai_response)
     response = transliterate_from_sinhala(response)
+    print(response)
+
     return response
 
 # Gradio Interface