Update app.py

app.py

@@ -2,7 +2,7 @@ import pandas as pd
 import spacy
 import gradio as gr
 import re
-
+import json
 
 dataset = pd.read_excel('Dataset-Verse-by-Verse.xlsx')
 dataset.rename(columns={'ArabicText': 'text'}, inplace=True)
@@ -35,8 +35,8 @@ def get_similar_sentences(text):
         similarity_score = ref_sentence.similarity(nlp(doc))
         similar_sentence = doc
         surah_name = dataset['SurahNameArabic'][i]
-        ayah_no = dataset['AyahNo'][i]
-        surah_no = dataset['SurahNo'][i]
+        ayah_no = int(dataset['AyahNo'][i])
+        surah_no = int(dataset['SurahNo'][i])
         similar_sentences.append({
             "similar_sentence": similar_sentence,
             "similarity_score": similarity_score,
@@ -46,7 +46,7 @@ def get_similar_sentences(text):
         })
     similar_sentences.sort(key=lambda x: x['similarity_score'], reverse=True)
     top_10 = similar_sentences[:10]
-    return top_10
+    return json.dumps(top_10)
 
 text_input = gr.inputs.Textbox(lines = 1 , label = "Enter a Quran Verse" )
 
@@ -57,5 +57,5 @@ examples = ['الحمدلله رب العالمين',
             'ونادى أصحاب الجنة أصحاب النار أن قد وجدنا ما وعدنا ربنا حقا فهل وجدتم ما وعد ربكم حقا ۖ قالوا نعم ۚ فأذن مؤذن بينهم أن لعنة الله على الظالمين'
             ]
 
-intf = gr.Interface(fn = get_similar_sentences , inputs = text_input , outputs = output_text , examples=examples )
+intf = gr.Interface(fn = get_similar_sentences , inputs = text_input , outputs = "json", examples=examples )
 intf.launch(debug = True)