Update app.py

app.py

@@ -15,14 +15,26 @@ def query_similarity(source_sentence, sentences):
         }
     }
     response = requests.post(API_URL, headers=headers, json=payload)
-    return response.json()
+    
+    # Ensure response is JSON
+    try:
+        return response.json(), sentences
+    except json.JSONDecodeError:
+        return {"error": "Failed to decode JSON response"}, sentences
 
-def format_output(response):
-    results = sorted(response, key=lambda x: x['score'], reverse=True)
-    formatted_results = []
-    for item in results:
-        formatted_results.append(f"Sentence: {item['sentence']}, Score: {item['score']:.4f}")
-    return "\n".join(formatted_results)
+def format_output(response, sentences):
+    if isinstance(response, list):
+        # Pair each score with its corresponding sentence
+        results = list(zip(response, sentences))
+        # Sort results by score in descending order
+        results = sorted(results, key=lambda x: x[0], reverse=True)
+        # Format the output
+        formatted_results = []
+        for score, sentence in results:
+            formatted_results.append(f"Sentence: {sentence}, Score: {score:.4f}")
+        return "\n".join(formatted_results)
+    else:
+        return f"Unexpected response format: {response}"
 
 def split_into_chunks(text, chunk_size=100):
     sentences = re.split(r'(?<=[.!?]) +', text)  # Split text into sentences
@@ -50,8 +62,8 @@ def semantic_search(query, file_path):
         with open(file_path, 'r', encoding='utf-8') as file:
             document = file.read()
         chunks = split_into_chunks(document)
-        response = query_similarity(query, chunks)
-        return format_output(response)
+        response, sentences = query_similarity(query, chunks)
+        return format_output(response, sentences)
     else:
         return "Please upload a .txt file."