Spaces:
Sleeping
Sleeping
File size: 3,023 Bytes
f2230da 6d604b5 669cdf3 201685d 669cdf3 f2230da 5011688 f2230da |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 |
from ultralytics import YOLO
import random
import streamlit as st
from PIL import Image
import os
import gdown
def extract_user_input(image=None):
id = "1e2pHejcp96q-NT9VmIDEY_hCA-tiFNef"
output = "trained_model.pt"
gdown.download(id = id, output = output, quiet=False)
model = YOLO(output)
result = model.predict(image)
names = model.names
det_cls = []
for r in result:
for c in r.boxes.cls:
det_cls.append(names[int(c)])
return det_cls[0]
def ai_choice():
choice = random.randint(0,1000000000)
if choice <=100000:
choice = "Rock"
elif choice >101 and choice <=1000000:
choice = "Paper"
else:
choice = "Scissors"
return choice
def winner(user_input,ai_input):
if user_input == "Rock":
if ai_input == "Scissors":
return "Player 1"
elif ai_input == "Paper":
return "AI"
else:
return "Draw!!!"
if user_input == "Paper":
if ai_input == "Rock":
return "Player 1"
elif ai_input == "Scissors":
return "AI"
else:
return "Draw!!!"
if user_input == "Scissors":
if ai_input == "Rock":
return "AI"
elif ai_input == "Paper":
return "Player 1"
else:
return "Draw!!!"
def main():
col1, col2 = st.columns(2)
with st.container():
st.header("Welcome to the Computer Vison powered Rock-Paper-Scissors Game!")
st.caption("developed by Indranil Bhattacharyya")
player_name = st.text_input("Please enter your name, Player!: ")
image = st.camera_input("Please upload your image: ")
if image:
st.image(image)
im = Image.open(image)
rgb_im = im.convert("RGB")
# exporting the image
rgb_im.save("user_input.jpg")
user_input = extract_user_input("user_input.jpg")
user_input_front_end = player_name + " has selected: " + user_input
st.code(user_input_front_end)
ai_choice_var = ai_choice()
ai_choice_front_end = "AI has selected: " + ai_choice_var
st.code(ai_choice_front_end)
if len(user_input)>0:
winner_name = winner(user_input,ai_choice_var)
if len(winner_name)!=0:
with st.container():
if winner_name == "Player 1":
winner_front_end = player_name + " Wins!!!"
st.success(winner_front_end)
else:
if winner_name == "AI":
winner_front_end = winner_name + " Wins!!!"
else:
winner_front_end = "It's a " + winner_name
st.error(winner_front_end)
else:
winner_name = ''
if __name__ == "__main__":
main()
|