Update app.py

app.py

@@ -42,8 +42,9 @@ def get_movie_info(movie_title):
             title = details_data.get("title", "Unknown 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}"
 
-            return f"Title: {title}, Year: {year}, Genre: {genre}"
+            return f"Title: {title}, Year: {year}, Genre: {genre}\nFind more info here: {tmdb_link}"
 
         else:
             return "Movie not found"
@@ -78,7 +79,7 @@ def generate_response(prompt):
     output = model.generate(**model_inputs, **gen_conf)
 
     generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
-    return generated_text
+    return f"Movie Info:\n{movie_info}\n\nGenerated Response:\n{generated_text}"
 
 # Create Gradio Interface
 iface = gr.Interface(fn=generate_response, inputs="text", outputs="text")