Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import os
+import multimodal_ai.py as ai
 
 examples = [
     {"text": "What is this image?", "files": ["sample/ducky.jpg"]},
@@ -20,8 +21,9 @@ def add_message(history, message):
 
 # Function to generate the bot's response
 def bot(history, user_input):
+    response = ai_assistant.generate_response(user_input)
     if history:
-        history[-1][1] = f"Cool! You said: {user_input}"
+        history[-1][1] = response
     return history
 
 def main():
@@ -57,4 +59,6 @@ def main():
     webApp.launch(share=True)
 
 if __name__ == "__main__":
+    # Initialize llm
+    ai_assistant = ai()
     main()