Update app.py

app.py

@@ -13,15 +13,9 @@ Equilibrium is the point where supply equals demand.
 Price acts as a signal for both producers and consumers.
 Market dynamics are influenced by shifts in supply and demand.
 """
-
 cleaned_text = supply_demand_text.strip()
 chunks = cleaned_text.split("\n")
-cleaned_chunks = []
-for chunk in chunks:
-    stripped_chunk = chunk.strip()
-    if stripped_chunk:
-        cleaned_chunks.append(stripped_chunk)
-
+cleaned_chunks = [chunk.strip() for chunk in chunks if chunk.strip()]
 chunk_embeddings = model.encode(cleaned_chunks, convert_to_tensor=True)
 
 def get_top_chunks(query):
@@ -29,7 +23,8 @@ def get_top_chunks(query):
     query_embedding = query_embedding / query_embedding.norm()
     normalized_chunks = chunk_embeddings / chunk_embeddings.norm(dim=1, keepdim=True)
     similarities = torch.matmul(normalized_chunks, query_embedding)
-    top_indices = torch.topk(similarities, k=3).indices
+    top_k = min(3, similarities.size(0))  # safer if fewer than 3 chunks
+    top_indices = torch.topk(similarities, k=top_k).indices
     top_chunks = [cleaned_chunks[i] for i in top_indices]
     return top_chunks