mrSpectrum commited on
Commit
23a3758
·
verified ·
1 Parent(s): f9d0953

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import gradio as gr
2
  from utils.bgg import get_game_details, get_hot_games, get_similar_games_v2, search
3
  from utils.wikipedia_tools import search_wiki, summary_wiki, page_wiki, random_wiki
 
4
 
5
 
6
  search_bgg = gr.Interface(
@@ -41,6 +42,8 @@ wiki_summary = gr.Interface(fn=summary_wiki, inputs=["text", "text"], outputs="t
41
  wiki_page = gr.Interface(fn=page_wiki, inputs=["text", "text"], outputs="text", title="Full Page", description="Get full content of a topic")
42
  wiki_random = gr.Interface(fn=random_wiki, inputs=["text"], outputs="text", title="Random Page", description="Get a random Wikipedia article")
43
 
 
 
44
  # ----- Combine All Tabs -----
45
  combined_tools = gr.TabbedInterface(
46
  [search_bgg, game_details, hot_games, recommend_games, wiki_search, wiki_summary, wiki_page, wiki_random],
 
1
  import gradio as gr
2
  from utils.bgg import get_game_details, get_hot_games, get_similar_games_v2, search
3
  from utils.wikipedia_tools import search_wiki, summary_wiki, page_wiki, random_wiki
4
+ from imdb import search_imdb
5
 
6
 
7
  search_bgg = gr.Interface(
 
42
  wiki_page = gr.Interface(fn=page_wiki, inputs=["text", "text"], outputs="text", title="Full Page", description="Get full content of a topic")
43
  wiki_random = gr.Interface(fn=random_wiki, inputs=["text"], outputs="text", title="Random Page", description="Get a random Wikipedia article")
44
 
45
+ imdb_interface = gr.Interface(fn=search_imdb, inputs="text", outputs="json", title="IMDb Movie Search", description="Get movie info")
46
+
47
  # ----- Combine All Tabs -----
48
  combined_tools = gr.TabbedInterface(
49
  [search_bgg, game_details, hot_games, recommend_games, wiki_search, wiki_summary, wiki_page, wiki_random],