Working with hardcoded stock_code value

app.py

@@ -30,21 +30,21 @@ def find_similar_items(stock_code):
     item_item_similarity_matrix['StockCode'] = CustomerID_Item_matrix.T.index
     item_item_similarity_matrix = item_item_similarity_matrix.set_index('StockCode')
 
-    # Randomly pick StockCode (22632) to display the most similar StockCode.
-    top_10_similar_items = list(
+    # Randomly pick StockCode (22865) to display the most similar StockCode.
+    top_5_similar_items = list(
         item_item_similarity_matrix\
-            .loc[22632]\
+            .loc[stock_code]\
             .sort_values(ascending=False)\
             .iloc[:5]\
         .index
     )
 
-    # Display the list of similar items of StockCode (23166) with item Description.
-    results_df = df1a.loc[df1a['StockCode'].isin(top_10_similar_items), ['StockCode', 'Description']].drop_duplicates().set_index('StockCode').loc[top_10_similar_items]
+    # Return the list of similar items of StockCode (23166) with item Description.
+    results_df = df1a.loc[df1a['StockCode'].isin(top_5_similar_items), ['StockCode', 'Description']].drop_duplicates().set_index('StockCode').loc[top_5_similar_items]
     return "\n" + results_df.to_string()
 
 # Set up the interface
 stock_code_input = gr.inputs.Textbox(label="Enter Stock Code:")
-output_table = gr.outputs.Textbox(label="Similar Items")
+output_table = gr.outputs.Textbox(label="Recommended Items")
 
-gr.Interface(fn=find_similar_items, inputs=stock_code_input, outputs=output_table, title="Similar Items Recommendation System", description="Enter a stock code to find similar items").launch()
+gr.Interface(fn=find_similar_items, inputs=stock_code_input, outputs=output_table).launch()