Update app.py

app.py

@@ -129,20 +129,27 @@ def predict(text):
     top_idx = int(util.pytorch_cos_sim(query_embedding, embeddings)[0].argmax())
     top_idx2 = int(util.pytorch_cos_sim(query_embedding, embeddings2)[0].argmax())
     
-    result = f"""Question 1: {df.iloc[top_idx]["question"]}
-Link 1: {df.iloc[top_idx]["link"]}
+#     result = f"""Question 1: {df.iloc[top_idx]["question"]}
+# Link 1: {df.iloc[top_idx]["link"]}
 
-Question 2: {df2.iloc[top_idx2]["question"]}
-Link 2: {df2.iloc[top_idx2]["link"]}"""
+# Question 2: {df2.iloc[top_idx2]["question"]}
+# Link 2: {df2.iloc[top_idx2]["link"]}"""
+    
+    # return result
     
-    return result
+    return {
+            "question1": df.iloc[top_idx]["question"],
+            "link1": df.iloc[top_idx]["link"],
+            "question2": df2.iloc[top_idx2]["question"],
+            "link2": df2.iloc[top_idx2]["link"]
+        }    
 
 # Match the EXACT structure of your working translation app
 title = "Search CSV"
 iface = gr.Interface(
     fn=predict,  # Changed from search_fatwa to predict 
     inputs=[gr.Textbox(label="text", lines=3)],
-    outputs='text',
+    outputs='json',
     title=title,
 )
 iface.launch()