Upload 3 files

app.py

@@ -0,0 +1,215 @@
+import gradio as gr 
+import pandas as pd 
+import numpy as np
+import random 
+import time
+import os
+from openai import OpenAI
+from dotenv import load_dotenv
+import requests
+from PIL import Image
+
+load_dotenv("saathi.env")
+
+client = OpenAI(api_key=os.environ["OPENAI_API_KEY"])
+
+def put_df2use(html):
+    df = pd.read_html(html)[0]
+    return df
+
+def make_msg_chunk_ai(history,cur_tem_selection,batting_team,batting_score,chase):
+    #hit flask here for system message with flag
+    req_params = {'flag':os.environ.get("FLAG")}
+    url = os.environ["FLASK_AI_PROMPT"]
+    response = requests.post(url, json=req_params)
+    full_data = response.json()
+    system_message=full_data['ai_data']
+    updated_system_message = system_message + f"\n Here is the senario by user \n First Batting team ->{batting_team} \n Batting Score ->{batting_score} \n Chase by second Team -> {chase} \n\n *Fantasy Team Selected By User : \n\n {cur_tem_selection.to_string(index=False)}\n\n--"
+    ai_msg = [{"role": "system", "content": updated_system_message}]
+    for chat in history:
+        user_m,bot_m=chat
+        if user_m:
+            val = {"role": "user", "content": user_m}
+            ai_msg.append(val)
+        if bot_m:
+            val = {"role": "assistant", "content": bot_m}
+            ai_msg.append(val)
+    return ai_msg
+
+def user(user_message, history):
+        return "", history + [[user_message, None]]
+
+def bot(history,cur_tem_selection,batting_team,batting_score,chase):
+    print(history,"====This is history")
+    bot_templete = make_msg_chunk_ai(history,cur_tem_selection,batting_team,batting_score,chase)
+    stream = client.chat.completions.create(
+    model="gpt-4-1106-preview",
+    messages=bot_templete,
+    stream=True)
+    history[-1][1] = ""
+    for chunk in stream:
+        if chunk.choices[0].delta.content is not None:
+            history[-1][1]+=chunk.choices[0].delta.content
+            yield history
+    
+
+def save_like_data(like_data: gr.LikeData):
+    print(like_data.liked,like_data.value,like_data.index)
+
+def activate_chat(chat,user_btn,clear_btn):
+    return gr.update(visible=True),gr.update(visible=True),gr.update(visible=True)
+
+def activate_full_builder_and_load_data(email):
+    #batting_score,chase,submit
+    if len(str(email))<5:
+        gr.Warning("Please Enter Valid Email")
+        return gr.update(visible=False),gr.update(visible=False),gr.update(visible=False),gr.update(visible=False),gr.update(visible=True),gr.update(visible=True)
+    request_params = {"email":email,"flag":os.environ.get("FLAG")}
+    url = os.environ["FLASK_API_ACCESS"]  
+
+    response = requests.post(url, json=request_params)
+    batting_choices = response.json().get("teams")
+    print(batting_choices,"response from flask")
+    return  gr.update(choices=batting_choices,visible=True),gr.update(visible=True),gr.update(visible=True),gr.update(visible=True),gr.update(visible=False),gr.update(visible=False)
+
+
+
+team_name_and_team_comp_dict={}
+def get_teams_data(batting_team,batting_score,chase):
+    # print(batting_team,"bat team",batting_score,"nan score",chase)
+    if batting_team and batting_score and chase and len(batting_team)>0 and len(batting_score)>0 and len(chase)>0:
+        req_params = {'batting_team':batting_team,
+                    'batting_score':batting_score,
+                    'chase':chase,
+                    'flag':os.environ.get("FLAG")}
+        url = os.environ["FLASK_TEAM_BUILDER"] 
+        gr.Warning("Making Teams")
+        response = requests.post(url, json=req_params)
+        
+        full_data = response.json()
+        
+        full_data=full_data['teams_generated']
+        team_name_and_team_comp_dict={}
+        all_teams_df =[]
+        for idx,team in enumerate(full_data):
+            team_name_and_team_comp_dict[f"Team {idx+1}"]=team
+            team = pd.DataFrame(team)
+            team["team_id"]=f"Team {idx+1}"
+            all_teams_df.append(team)
+        all_teams_df = pd.concat(all_teams_df)
+        all_team_name = list(team_name_and_team_comp_dict.keys())
+        first_df = pd.DataFrame(team_name_and_team_comp_dict[all_team_name[0]])
+        first_html = first_df.sample(len(first_df)).to_html(classes='ui celled table', index=False)
+        
+        return gr.update(choices=all_team_name,value=all_team_name[0],visible=True,interactive=True),gr.update(value=first_html,visible=True),gr.update(visible=True),gr.update(visible=True),all_teams_df
+    else:
+        gr.Warning("Please Verify Inputs")
+        # return gr.update(visible=False,interactive=True),gr.update(visible=False),gr.update(visible=False),gr.update(visible=False),pd.DataFrame()
+        return None
+    
+def pivot_role_team_with_team_names(df):
+    # Copying the original dataframe to not alter it
+    df_copy = df.copy()
+    
+    # Defining the order for the role_type
+    role_order = ["Wicket-keeper", "Batsman", "All-rounder", "Bowler"]
+    
+    # Pivoting the dataframe
+    pivot_df = df_copy.pivot_table(index='role_type', columns='team', aggfunc='size', fill_value=0)
+    
+    # Reindex the pivot table to ensure the order of role_types
+    pivot_df = pivot_df.reindex(role_order)
+    
+    # Adding a total row
+    pivot_df.loc['Total', :] = pivot_df.sum()
+    
+    # Resetting index to make 'role_type' a column instead of an index and renaming it for clarity
+    pivot_df.reset_index(inplace=True)
+    pivot_df.rename(columns={'role_type': 'Role'}, inplace=True)
+    pivot_df.rename_axis(None, axis=1, inplace=True)
+    
+    return pivot_df
+
+def fix_hit(hit):
+    try:
+        hit = float(hit)
+        hit = round(hit,1)
+    except Exception as e:
+        hit = ""
+    return hit
+        
+
+def display_selected_df(team_index,all_teams_df):
+    temp_df = all_teams_df[all_teams_df["team_id"]==team_index]
+    temp_df = temp_df.drop(["team_id","player_code"],axis=1)
+    temp_df2 = temp_df[["player_name","role_type","team","hit"]]
+    temp_df = temp_df[["player_name","role_type","team"]]
+    disp_df = pivot_role_team_with_team_names(temp_df)
+    temp_df2["hit"]=temp_df2["hit"].apply(fix_hit)
+    temp_df2.columns=["Player Name","Role","Team","HIT%"]
+    # html_header = f"<h2>{team_index}</h2>"
+    req_html= temp_df2.sample(len(temp_df2)).to_html(classes='ui celled table', index=False)
+    print("changing data")
+    
+    return req_html,disp_df
+    
+    
+intro_html='''<h3> Welcome to AI-Saathi </h3>
+<p>Saare fantasy problems ka solution</p>
+'''
+
+with gr.Blocks(theme=gr.themes.Default(primary_hue="amber", secondary_hue="pink")) as demo:
+    # gr.HTML(intro_html)
+    logo = Image.open("assets/IMG_0453.jpg")
+    gr.Image(logo)
+    with gr.Row():
+        email_input = gr.Textbox(label="Email ID",placeholder="Enter Email for access. Querries : www.apnacricketteam.com",lines=1)
+    with gr.Row():
+        activation_btn=gr.Button("Load Team Data")    
+    
+    with gr.Row():
+        batting_team = gr.Dropdown(choices=[], label="Batting Team",visible=False,interactive=True)
+        batting_score = gr.Dropdown(choices=["Low", "Average", "High"], label="Batting Score",visible=False,interactive=True)
+        chase = gr.Dropdown(choices=["Yes", "No"], label="Chased",visible=False,interactive=True)
+        all_teams_df = gr.DataFrame(visible=False)
+    with gr.Row():
+        submit = gr.Button("Submit",visible=False)
+    with gr.Row():
+        team_dropdown = gr.Dropdown(label="Select Team",allow_custom_value=False,multiselect=False,visible=False)
+    with gr.Row():
+        html_data = gr.HTML(visible=False)
+    with gr.Row():
+        basic_status = gr.DataFrame(label="Team Stats",visible=False)
+    with gr.Row():
+        btn = gr.Button("Team improvement with AI-Saathi",visible=False)
+        curr_team_data=gr.DataFrame(visible=False,interactive=False)
+    with gr.Row():
+        chat_bot = gr.Chatbot(visible=False)
+        thread_id = gr.Textbox(value="THIS IS THE THREAD-ID",visible=False)
+    with gr.Row():
+        user_msg = gr.Textbox(placeholder="Team ko change ya improve karne ke liye suggestion le",visible=False)
+    with gr.Row():
+        clear_btn = gr.ClearButton(chat_bot,visible=False)
+        
+    activation_btn.click(fn=activate_full_builder_and_load_data,inputs=[email_input,],outputs=[batting_team,batting_score,chase,submit,email_input,activation_btn])
+    submit.click(fn=get_teams_data,inputs=[batting_team,batting_score,chase],outputs=[team_dropdown,html_data,basic_status,btn,all_teams_df])
+    team_dropdown.change(fn=display_selected_df, inputs=[team_dropdown,all_teams_df], outputs=[html_data,basic_status])
+    
+    
+    html_data.change(fn=put_df2use,inputs=html_data,outputs=curr_team_data)
+    
+    btn.click(fn=activate_chat,inputs=[chat_bot,user_msg,clear_btn],outputs=[chat_bot,user_msg,clear_btn])
+    
+    user_msg.submit(user, [user_msg, chat_bot], [user_msg, chat_bot], queue=True).then(
+        bot, [chat_bot,curr_team_data,batting_team,batting_score,chase], chat_bot)
+    
+    clear_btn.click(lambda: None, None, chat_bot, queue=False)
+    team_dropdown.change(lambda: None, None,chat_bot)
+    team_dropdown.change(lambda: None, None,thread_id)
+    chat_bot.like(fn=save_like_data)
+    
+
+demo.launch(share=False)
+
+    
+

requirements.txt

@@ -0,0 +1,113 @@
+aiofiles==23.2.1
+altair==5.2.0
+annotated-types==0.6.0
+anyio==4.3.0
+appnope==0.1.4
+asttokens==2.4.1
+attrs==23.2.0
+cachetools==5.3.2
+certifi==2024.2.2
+charset-normalizer==3.3.2
+click==8.1.7
+colorama==0.4.6
+comm==0.2.1
+contourpy==1.2.0
+cycler==0.12.1
+debugpy==1.8.1
+decorator==5.1.1
+distro==1.9.0
+exceptiongroup==1.2.0
+executing==2.0.1
+fastapi==0.110.0
+ffmpy==0.3.2
+filelock==3.13.1
+fonttools==4.49.0
+fsspec==2024.2.0
+google-api-core==2.17.1
+google-api-python-client==2.119.0
+google-auth==2.28.1
+google-auth-httplib2==0.2.0
+google-auth-oauthlib==1.2.0
+googleapis-common-protos==1.62.0
+gradio==4.19.2
+gradio_client==0.10.1
+h11==0.14.0
+httpcore==1.0.4
+httplib2==0.22.0
+httpx==0.27.0
+huggingface-hub==0.20.3
+idna==3.6
+importlib-metadata==7.0.1
+importlib-resources==6.1.1
+ipykernel==6.29.2
+ipython==8.18.1
+jedi==0.19.1
+Jinja2==3.1.3
+jsonschema==4.21.1
+jsonschema-specifications==2023.12.1
+jupyter_client==8.6.0
+jupyter_core==5.7.1
+kiwisolver==1.4.5
+lxml==5.1.0
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+matplotlib==3.8.3
+matplotlib-inline==0.1.6
+mdurl==0.1.2
+nest-asyncio==1.6.0
+numpy==1.26.4
+oauthlib==3.2.2
+openai==1.13.3
+orjson==3.9.15
+packaging==23.2
+pandas==2.2.1
+parso==0.8.3
+pexpect==4.9.0
+pillow==10.2.0
+platformdirs==4.2.0
+prompt-toolkit==3.0.43
+protobuf==4.25.3
+psutil==5.9.8
+ptyprocess==0.7.0
+pure-eval==0.2.2
+pyasn1==0.5.1
+pyasn1-modules==0.3.0
+pydantic==2.6.2
+pydantic_core==2.16.3
+pydub==0.25.1
+Pygments==2.17.2
+pygsheets==2.0.6
+pyparsing==3.1.1
+python-dateutil==2.8.2
+python-dotenv==1.0.1
+python-multipart==0.0.9
+pytz==2024.1
+PyYAML==6.0.1
+pyzmq==25.1.2
+referencing==0.33.0
+requests==2.31.0
+requests-oauthlib==1.3.1
+rich==13.7.0
+rpds-py==0.18.0
+rsa==4.9
+ruff==0.2.2
+semantic-version==2.10.0
+shellingham==1.5.4
+six==1.16.0
+sniffio==1.3.0
+stack-data==0.6.3
+starlette==0.36.3
+tomlkit==0.12.0
+toolz==0.12.1
+tornado==6.4
+tqdm==4.66.2
+traitlets==5.14.1
+typer==0.9.0
+typing_extensions==4.9.0
+tzdata==2024.1
+uritemplate==4.1.1
+urllib3==2.2.1
+uvicorn==0.27.1
+wcwidth==0.2.13
+websockets==11.0.3
+zipp==3.17.0