Update app.py

app.py

@@ -43,12 +43,12 @@ def get_movie_info(movie_title):
             year = details_data.get("release_date", "Unknown Year")[:4]
             genre = ", ".join(genre["name"] for genre in details_data.get("genres", []))
             tmdb_link = f"https://www.themoviedb.org/movie/{movie_id}"
-          #  poster_path = details_data.get("poster_path")
+            poster_path = details_data.get("poster_path")
             
             # Convert poster_path to a complete image URL
-            # image_url = f"https://image.tmdb.org/t/p/w500{poster_path}" if poster_path else ""
+            image_url = f"https://image.tmdb.org/t/p/w500{poster_path}" if poster_path else ""
 
-            return f"Title: {title}, Year: {year}, Genre: {genre}\nFind more info here: {tmdb_link}"
+            return f"Title: {title}, Year: {year}, Genre: {genre}\nFind more info here: {tmdb_link}", image_url
 
         else:
             return "Movie not found", ""
@@ -65,7 +65,7 @@ def generate_response(prompt):
     )
 
     # Call the get_movie_info function to enrich the response
-    movie_info = get_movie_info(prompt)
+    movie_info, image_url = get_movie_info(prompt)
 
     # Concatenate the movie info with the input template
     input_text_template += f" Movie Info: {movie_info}"
@@ -83,7 +83,7 @@ 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}"
+    return f"Movie Info:\n{movie_info}\n\nGenerated Response:\n{generated_text}\nImage URL: {image_url}"
 
 # Define chat function for gr.ChatInterface
 def chat_function(message, history):
@@ -93,4 +93,4 @@ def chat_function(message, history):
 
 # Create Gradio Chat Interface
 chat_interface = gr.ChatInterface(chat_function)
-chat_interface.launch()
+chat_interface.launch()