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}", image_url
+            return f"Title: {title}, Year: {year}, Genre: {genre}\nFind more info here: {tmdb_link}"
 
         else:
             return "Movie not found", ""
@@ -83,9 +83,14 @@ 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}", image_url
+    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", "image"])
-iface.launch()
+# Define chat function for gr.ChatInterface
+def chat_function(message, history):
+    response = generate_response(message)
+    history.append([message, response])
+    return response
 
+# Create Gradio Chat Interface
+chat_interface = gr.ChatInterface(chat_function)
+chat_interface.launch()