Update app.py

app.py

@@ -2,7 +2,6 @@ import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import gradio as gr
 import requests
-from IPython.display import display, Image
 
 model_name = "Writer/palmyra-small"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
@@ -44,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}", image_url
+            return f"Title: {title}, Year: {year}, Genre: {genre}\nFind more info here: {tmdb_link}"
 
         else:
             return "Movie not found", ""
@@ -66,7 +65,7 @@ def generate_response(prompt):
     )
 
     # Call the get_movie_info function to enrich the response
-    movie_info, image_url = get_movie_info(prompt)
+    movie_info = get_movie_info(prompt)
 
     # Concatenate the movie info with the input template
     input_text_template += f" Movie Info: {movie_info}"
@@ -85,10 +84,10 @@ def generate_response(prompt):
 
     generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
     
-    # Embed image directly in the response
-    display(Image(url=image_url, alt="Movie Poster"))
+    # Display image directly in the chat
+    # image_component = gr.Image(image_url)
     
-    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}\n"
 
 # Define chat function for gr.ChatInterface
 def chat_function(message, history):
@@ -100,3 +99,4 @@ def chat_function(message, history):
 chat_interface = gr.ChatInterface(chat_function)
 chat_interface.launch(share=True)  # Added share=True to create a public link
 
+