Update app.py

app.py

@@ -1,36 +1,46 @@
-import streamlit as st
-from transformers import AutoModelForCausalLM, AutoTokenizer
-from accelerate import Accelerator  # Import Accelerator from the accelerate library
-
-def main():
-    st.title("Chatbot with Hugging Face Model")
-
-    model_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
-    tokenizer = AutoTokenizer.from_pretrained(model_id)
-    
-    # Create an Accelerator instance
-    accelerator = Accelerator()
-
-    # Use the Accelerator for initializing the model
-    model = AutoModelForCausalLM.from_pretrained(model_id, device_map=accelerator.device)
-
-    user_input = st.text_input("User Input:", "What is your favourite condiment?")
-
-    if st.button("Generate Response"):
-        messages = [
-            {"role": "user", "content": user_input},
-            {"role": "assistant", "content": "Placeholder assistant message"}
-        ]
-
-        inputs = tokenizer.apply_chat_template(messages, return_tensors="pt").to(accelerator.device)
-
-        # Use the Accelerator for generating outputs
-        with accelerator.device():
-            outputs = model.generate(inputs, max_new_tokens=20)
-
-        response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-
-        st.text_area("Assistant's Response:", response)
+def print_board(board):
+    for row in board:
+        print(" | ".join(row))
+        print("-" * 9)
+
+def check_winner(board, player):
+    # Check rows, columns, and diagonals
+    for i in range(3):
+        if all(cell == player for cell in board[i]) or all(board[j][i] == player for j in range(3)):
+            return True
+    if all(board[i][i] == player for i in range(3)) or all(board[i][2 - i] == player for i in range(3)):
+        return True
+    return False
+
+def check_tie(board):
+    return all(cell != " " for row in board for cell in row)
+
+def tic_tac_toe():
+    board = [[" " for _ in range(3)] for _ in range(3)]
+    current_player = "X"
+
+    print("Welcome to Tic-Tac-Toe!")
+
+    while True:
+        print_board(board)
+
+        row = int(input(f"Player {current_player}, enter row (0, 1, or 2): "))
+        col = int(input(f"Player {current_player}, enter column (0, 1, or 2): "))
+
+        if board[row][col] == " ":
+            board[row][col] = current_player
+            if check_winner(board, current_player):
+                print_board(board)
+                print(f"Player {current_player} wins! Congratulations!")
+                break
+            elif check_tie(board):
+                print_board(board)
+                print("It's a tie! Well played.")
+                break
+            else:
+                current_player = "O" if current_player == "X" else "X"
+        else:
+            print("Cell already occupied. Try again.")
 
 if __name__ == "__main__":
-    main()
+    tic_tac_toe()