Upload app.py

app.py

@@ -115,10 +115,15 @@ def gradio_search(query, game_name, selected_model, aggregator, pool_size, k=6):
   
   results = []
   relevant_videos = searcher.search_top_k(query, k=k, pool_size=pool_size, search_mod=aggregator)
+  
+  
   params = ', '.join(map(str, [query, game_name, selected_model, aggregator, pool_size]))
   results.append(params)
-  results.extend(relevant_videos)
-  print(results)
+  
+  for v in relevant_videos:
+    results.append(v)
+    sid = v.split('/')[-1].split('.')[0]
+    results.append(f"https://www.reddit.com/r/GamePhysics/comments/{sid}/")
   return results
 
 list_of_games = ['Grand Theft Auto V']
@@ -141,10 +146,16 @@ iface = gr.Interface(fn=gradio_search,
                      outputs=[
                               gr.outputs.Textbox(type="auto", label='Search Params'),
                               gr.outputs.Video(type='mp4', label='Result 1'),
+                              gr.outputs.Textbox(type="auto", label='Submission URL - Result 1'),
                               gr.outputs.Video(type='mp4', label='Result 2'),
+                              gr.outputs.Textbox(type="auto", label='Submission URL - Result 2'),
                               gr.outputs.Video(type='mp4', label='Result 3'),
+                              gr.outputs.Textbox(type="auto", label='Submission URL - Result 3'),
                               gr.outputs.Video(type='mp4', label='Result 4'),
-                              gr.outputs.Video(type='mp4', label='Result 5')],
+                              gr.outputs.Textbox(type="auto", label='Submission URL - Result 4'),
+                              gr.outputs.Video(type='mp4', label='Result 5'),
+                              gr.outputs.Textbox(type="auto", label='Submission URL - Result 5'),
+                              ],
                      examples=[
                                ["A red car", list_of_games[0], 'ViT-B/32', 'Top-K', 1000],
                                ["A person wearing pink", list_of_games[0], 'ViT-B/32', 'Top-K', 1000],