Update app.py

app.py

@@ -43,14 +43,18 @@ 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")
+            
+            # Convert poster_path to a complete image URL
+            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"
-    
+            return "Movie not found", ""
+
     except Exception as e:
-        return f"Error: {e}"
+        return f"Error: {e}", ""
 
 def generate_response(prompt):
     input_text_template = (
@@ -61,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}"
@@ -79,8 +83,9 @@ 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}", image_url
 
 # Create Gradio Interface
-iface = gr.Interface(fn=generate_response, inputs="text", outputs="text")
+iface = gr.Interface(fn=generate_response, inputs="text", outputs=["text", "image"])
 iface.launch()
+