Update app.py

app.py

@@ -1,3 +1,5 @@
+# app.py (Finance Chatbot)
+
 import gradio as gr
 from helper import get_together_api_key
 from guardrail import is_safe
@@ -6,27 +8,11 @@ from together import Together
 # Initialize Together client
 client = Together(api_key=get_together_api_key())
 
-# Gradio interface functions
+# Function to handle the chatbot's response to user queries
 def run_action(message, history):
-    global game_state, game_running  # Access the global game state and game status
-
-    if not game_running:
-        return "The game has ended. Type 'restart the game' to play again."
-
-    if message.lower() == "start game":
-        return game_state["start"]
-
-    if message.lower() == "restart the game":
-        game_state = initialize_game_state()
-        return "Game restarted! " + game_state["start"]
-
-    if message.lower() == "exit":
-        game_running = False
-        return "The game has ended. Type 'restart the game' to play again."
-
     system_prompt = """You are a financial assistant. You can only answer finance-related queries.
     - Do not answer non-finance questions.
-    - Answer in 50 words
+    - Answer in 50 words.
     - Ensure responses adhere to the safety policy."""
 
     messages = [{"role": "system", "content": system_prompt}]
@@ -49,7 +35,7 @@ def run_action(message, history):
 
     return model_output.choices[0].message.content
 
-
+# Main loop for the chatbot to handle user input
 def main_loop(message, history):
     """
     Main loop for the chatbot to handle user input.
@@ -61,7 +47,6 @@ def main_loop(message, history):
     # Generate and validate the response
     return run_action(message, history)
 
-
 # Gradio Chat Interface
 demo = gr.ChatInterface(
     main_loop,
@@ -82,4 +67,4 @@ demo = gr.ChatInterface(
 )
 
 # Launch the Gradio app
-demo.launch(share=True, server_name="0.0.0.0")
+demo.launch(share=True, server_name="0.0.0.0")