Update app.py

app.py

@@ -9,11 +9,12 @@ from classes.Qlearningagent import QlearningAgent
 
 def main():
   custom_css = """
+    #color {background-color: #00BFFF}
     .gradio-button {
       width: 75px; /* Defina a largura desejada */
       height: 75px; /* Defina a altura desejada */
-      background-color: red; /* Cor de fundo azul */
-      color: white; /* Cor do texto (branco neste caso) */
+      background-color: #6495ED; /* Cor de fundo azul */
+      color: black; /* Cor do texto (branco neste caso) */
       border: none; /* Remove a borda */
       text-align: center; /* Centraliza o texto horizontalmente */
       text-decoration: none; /* Remove a decoração do texto */
@@ -32,13 +33,12 @@ def main():
         board.move(i,j,int(usr)) 
         w = board.check_win()
         temp-=1
-        return gr.update(size='lg', scale = 0, min_width = 100, value=usr_piece, interactive=True), gr.Number(value=temp, visible=False), gr.Number(value=w, visible=False), board
+        return gr.update(size='lg',elem_id='color', scale = 0, min_width = 100, value=usr_piece, interactive=True), gr.Number(value=temp, visible=False), gr.Number(value=w, visible=False), board
       else:
-        return gr.update(size='lg', scale = 0, min_width = 100, interactive=True), gr.update(value=temp, visible=False),gr.update(visible=False)
+        return gr.update(size='lg',elem_id='color', scale = 0, min_width = 100, interactive=True), gr.update(value=temp, visible=False),gr.update(visible=False)
       
           
   def ia_move(board,game_over,ia):
-     
       if int(game_over) == -1:
         state_board = board.matriz.copy()
         avaible_moves = board.get_avaible_moves()
@@ -48,43 +48,46 @@ def main():
         board.move(i,j,int(ia))
         w = board.check_win()
         #buttons
-        retornos_btn = [gr.Button(size='lg', scale = 0, min_width = 100, interactive=True)]*(action+1)
-        retornos_btn[-1] = gr.Button(size='lg', scale = 0, min_width = 100, interactive=True, value=ia_piece)
-        retornos_btn.extend([gr.Button(size='lg', scale = 0, min_width = 100,  interactive=True)]*(9 - action-1))
+        retornos_btn = [gr.Button(size='lg',elem_id='color', scale = 0, min_width = 100, interactive=True)]*(action+1)
+        retornos_btn[-1] = gr.Button(size='lg',elem_id='color', scale = 0, min_width = 100, interactive=True, value=ia_piece)
+        retornos_btn.extend([gr.Button(size='lg',elem_id='color', scale = 0, min_width = 100,  interactive=True)]*(9 - action-1))
         retornos_btn.append(gr.Number(value=w, visible=False))
         retornos_btn.append(board)
 
         return retornos_btn
       else:
-        retornos_btn = [gr.Button(size='lg', scale = 0, min_width = 100, interactive=True)]*9
+        retornos_btn = [gr.Button(size='lg',elem_id='color', scale = 0, min_width = 100, interactive=True)]*9
         retornos_btn.append(gr.Number(visible=False))
         retornos_btn.append(board)
 
         return retornos_btn
           
   def set_col_visible(board,game_over,temp,usr,ia,selected):
-      game_over = -1
       usr =  1 if selected  == 'X' else 2
       ia = 2 if usr == 1 else 1
-      retorn_buttons = [gr.Button(size='lg', scale = 0, min_width = 100,value='',elem_classes='gradio-button',interactive=True)]*9
+      game_over = -1
+      retorn_buttons = [gr.Button(size='lg',elem_id='color', scale = 0, min_width = 100,value='',elem_classes='gradio-button',interactive=True)]*9
       if selected == 'X':
         retorn_buttons.append(gr.Number(value=temp, visible=False))
         retorn_buttons.append(gr.Number(value=-1, visible=False))
         retorn_buttons.append(gr.Number(value=ia, visible=False))
         retorn_buttons.append(gr.Number(value=usr, visible=False))
+        
         retorn_buttons.append(board)
+
       else:
         retorn_buttons = ia_move(board,game_over,ia)
         retorn_buttons.insert(-1,gr.Number(value=-1, visible=False))
         retorn_buttons.insert(-1,gr.Number(value=ia, visible=False))
         retorn_buttons.insert(-1,gr.Number(value=usr, visible=False))
       
+
       return retorn_buttons
       
           
   def reset(board):
       board.reset_matrix()
-      buttons = [gr.Button(size='lg', scale = 0, min_width = 100,elem_classes='gradio-button',interactive=False)]*9
+      buttons = [gr.Button(size='lg',elem_id='color', scale = 0, min_width = 100,elem_classes='gradio-button',interactive=False)]*9
       buttons.append(board)
       return buttons
 
@@ -108,7 +111,7 @@ def main():
       return gr.update(value=msg, show_label=False), gr.Number(value=int(game_over), visible=False)
   
   parser = argparse.ArgumentParser()
-  parser.add_argument('--file_name', type=str, required=False, default='models/q_agent-0.5ep-0.6ap-0.9-1000000r.json')
+  parser.add_argument('--file_name', type=str, required=False, default='models/q_agent-0.9ep-0.6ap-0.9-1000r.json')
   args = parser.parse_args()
   q_agent= QlearningAgent(epsilon=0,alpha=0,discount_factor=0, train=False)
   q_agent.load_agent_dict(args.file_name)
@@ -118,17 +121,17 @@ def main():
         with gr.Column() as x:
           title = gr.Text(value='Escolha uma peça', show_label=False)
           with gr.Row():
-              p1 = gr.Button(size='lg', scale = 0, min_width = 100,value='',elem_classes='gradio-button',interactive=False)
-              p2 = gr.Button(size='lg', scale = 0, min_width = 100,value='',elem_classes='gradio-button',interactive=False)
-              p3 = gr.Button(size='lg', scale = 0, min_width = 100,value='',elem_classes='gradio-button',interactive=False)
+              p1 = gr.Button(size='lg',elem_id='color', scale = 0, min_width = 100,value='',elem_classes='gradio-button',interactive=False)
+              p2 = gr.Button(size='lg',elem_id='color', scale = 0, min_width = 100,value='',elem_classes='gradio-button',interactive=False)
+              p3 = gr.Button(size='lg',elem_id='color', scale = 0, min_width = 100,value='',elem_classes='gradio-button',interactive=False)
           with gr.Row():
-              p4 = gr.Button(size='lg', scale = 0, min_width = 100,value='',elem_classes='gradio-button',interactive=False)
-              p5 = gr.Button(size='lg', scale = 0, min_width = 100,value='',elem_classes='gradio-button',interactive=False)
-              p6 = gr.Button(size='lg', scale = 0, min_width = 100,value='',elem_classes='gradio-button',interactive=False)
+              p4 = gr.Button(size='lg',elem_id='color', scale = 0, min_width = 100,value='',elem_classes='gradio-button',interactive=False)
+              p5 = gr.Button(size='lg',elem_id='color', scale = 0, min_width = 100,value='',elem_classes='gradio-button',interactive=False)
+              p6 = gr.Button(size='lg',elem_id='color', scale = 0, min_width = 100,value='',elem_classes='gradio-button',interactive=False)
           with gr.Row():
-              p7 = gr.Button(size='lg', scale = 0, min_width = 100,value='',elem_classes='gradio-button',interactive=False)
-              p8 = gr.Button(size='lg', scale = 0, min_width = 100,value='',elem_classes='gradio-button',interactive=False)
-              p9 = gr.Button(size='lg', scale = 0, min_width = 100,value='',elem_classes='gradio-button',interactive=False)
+              p7 = gr.Button(size='lg',elem_id='color', scale = 0, min_width = 100,value='',elem_classes='gradio-button',interactive=False)
+              p8 = gr.Button(size='lg',elem_id='color', scale = 0, min_width = 100,value='',elem_classes='gradio-button',interactive=False)
+              p9 = gr.Button(size='lg',elem_id='color', scale = 0, min_width = 100,value='',elem_classes='gradio-button',interactive=False)
 
         n1 = gr.Number(value="1", visible=False)
         n2 = gr.Number(value="2", visible=False)