Update app.py

app.py

@@ -83,14 +83,19 @@ def generate_response(prompt):
     output = model.generate(**model_inputs, **gen_conf)
 
     generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
-    return f"Movie Info:\n{movie_info}\n\nGenerated Response:\n{generated_text}\nImage URL: {image_url}"
+    
+    # Display image directly in the chat
+    image_component = gr.Image(image_url, alt="Movie Poster")
+    
+    return f"Movie Info:\n{movie_info}\n\nGenerated Response:\n{generated_text}\n", image_component
 
 # Define chat function for gr.ChatInterface
 def chat_function(message, history):
-    response = generate_response(message)
+    response, image_component = generate_response(message)
     history.append([message, response])
-    return response
+    return response, image_component
 
 # Create Gradio Chat Interface
 chat_interface = gr.ChatInterface(chat_function)
 chat_interface.launch()
+