Weekend work on the gradio interface

app.py

@@ -1,26 +1,49 @@
 import gradio as gr
+import pandas as pd
 import sys
 sys.path.append("rd2l_pred")
 from training_data_prep import list_format, modification, league_money, df_gen
+from feature_engineering import heroes, hero_information
 
-def greet(name):
-    return "Hello " + name + "!!"
 
+def fetch_data(user_id, mmr, comf_1, comf_2, comf_3, comf_4, comf_5):
+    player_id = user_id.split("/")[-1]
+
+    series = {"player_id" : player_id, "mmr" : mmr, "p1" : comf_1, "p2" : comf_2, "p3" : comf_3, "p4" : comf_4, "p5" : comf_5}
+
+    money = pd.read_csv("result_money.csv")    
+
+    print(money)
+    print(money.values)
+
+    for item in money.values:
+        series.update({item[0] : float(item[1])})
+
+    print("Corrected Series")
+    print(series)
+    
+    d = pd.Series(series)
+    print(d)
+
+    # This the original section used to add the money section to the series.
+    # d = d.assign(count=lambda x: money[player_season].loc['count'], mean=lambda x: money[player_season].loc['mean'], std=lambda x: money[player_season].loc['std'], min=lambda x: money[player_season].loc['min'], max=lambda x: money[player_season].loc['max'], sum=lambda x: money[player_season].loc['sum'])
 
-def fetch_data(user_input):
     # We need to generate the inputs for the sheet using hugging face
     # We also need to load the money values from the generated csv file
-    df_gen(draft, league_money(captains, data_type), data_type)
-    return
-
-# Needs a text input for dotabuff link
+    # df_gen(draft, league_money(captains, data_type), data_type)
 
+    return player_id
 
-demo = gr.Interface(fn=greet, inputs="textbox", outputs="text")
 demo = gr.Interface(
-    fn=greet,
+    fn=fetch_data,
     inputs=[
-        "textbox"
+        gr.Textbox(label="Player ID or Link to OpenDota/Dotabuff"),
+        gr.Textbox(label="MMR"),
+        gr.Slider(1, 5, value=1, step=1, label="Comfort (Pos 1)"),
+        gr.Slider(1, 5, value=1, step=1, label="Comfort (Pos 2)"),
+        gr.Slider(1, 5, value=1, step=1, label="Comfort (Pos 3)"),
+        gr.Slider(1, 5, value=1, step=1, label="Comfort (Pos 4)"),
+        gr.Slider(1, 5, value=1, step=1, label="Comfort (Pos 5)")
         # gr.Image(type="filepath"),
         # gr.Dropdown(["Pinch", "Spiral", "Shift Up", "Bulge", "Volcano"], value="Bulge", label="Function"), 
         # gr.Checkbox(label="Randomize inputs?"),