Update app.py

app.py

@@ -12,16 +12,6 @@ load_dotenv()
 # Now you can create a Groq client without passing the api_key
 client = Groq(api_key=os.getenv('GROQ_API_KEY'))
 
-
-# from langchain_community.chat_models import ChatOpenAI
-# from langchain.schema import AIMessage, HumanMessage
-from dotenv import load_dotenv
-load_dotenv()
-
-client = Groq(
-    api_key=os.environ.get("GROQ_API_KEY"),
-)
-
 SYSTEM_MSG = """
 You are Jeff Bezos, founder of Amazon and Blue Origin, known for your visionary leadership, customer obsession, and relentless pursuit of innovation. You have a deep understanding of business strategy, e-commerce, technology, and space exploration. You communicate with clarity, confidence, and forward-thinking insights, often encouraging bold thinking and long-term planning. You approach problems with data-driven reasoning, always considering the impact on customers and the future. Your tone is professional, thoughtful, and occasionally humorous, with a focus on inspiring others to think big and take calculated risks."""
 
@@ -70,29 +60,18 @@ def predict(message, history):
             yield partialMessage
 
     pprint(f"[Tokens={chunkCount}] {message}")
-    
-
-
 
 # Define the Chatbot component and store it in a variable
 chatbot_component = gr.Chatbot(label="Business Advice from Jeff Bezos")
 
-with gr.ChatInterface(
-    predict,
-    title="Business Advice from Jeff Bezos ",
-    theme=gr.themes.Soft(), 
-    chatbot=chatbot_component,
-    textbox=gr.Textbox(
-        placeholder="Ask Your Question",
-        scale=7,
-        max_lines=2,
-    )
-) as demo:
-    demo.load(initial_message, None, chatbot_component)  # Pass the actual component here
-    demo.load(__attachIp, None, None)
-
-demo.launch(debug=True)
-
-
+demo = gr.Interface(
+    fn=predict,
+    inputs="text",
+    outputs="text",
+    title="Business Advice from Jeff Bezos",
+    theme=gr.themes.Soft(),
+)
 
+demo.set_on_chat_load(__attachIp)  # Attach IP handler
 
+demo.launch(debug=True)