Merge branch 'master' into main

.gitignore

@@ -0,0 +1,3 @@
+__pycache__/
+.env
+

Dockerfile

@@ -0,0 +1,20 @@
+# Use an official Python runtime as a parent image
+FROM python:3.8-slim-buster
+
+# Set the working directory in the container to /app
+WORKDIR /app
+
+# Copy the current directory contents into the container at /app
+ADD . /app
+
+# Install any needed packages specified in requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Make port 7860 available to the world outside this container
+EXPOSE 7860
+
+# Define environment variable
+ENV NAME World
+
+# Run app.py when the container launches
+CMD ["python", "app.py"]

agent.py

@@ -0,0 +1,70 @@
+
+import re
+import json
+
+import functools
+import functions
+import torch
+from tools import tools
+from transformers import (
+    AutoModelForCausalLM,
+    AutoTokenizer,
+    BitsAndBytesConfig
+)
+
+# Get all functions from functions.py
+all_functions = [func for func in dir(functions) if callable(
+    getattr(functions, func)) and not func.startswith("__")]
+
+# Create names_to_function dict containing partials for all functions in functions.py
+names_to_functions = {func: functools.partial(
+    getattr(functions, func)) for func in all_functions}
+
+
+model_id = "meta-llama/Meta-Llama-3.1-8B-Instruct"
+
+# specify how to quantize the model
+quantization_config = BitsAndBytesConfig(
+    load_in_4bit=True,
+    bnb_4bit_quant_type="nf4",
+    bnb_4bit_compute_dtype=torch.bfloat16,
+)
+
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+model = AutoModelForCausalLM.from_pretrained(
+    model_id, device_map="auto", quantization_config=quantization_config
+)
+
+
+def extract_function_call(output):
+    match = re.search(r'<\|python_tag\|>(.*)<\|eom_id\|>', output)
+    if match:
+        function_call = match.group(1)
+        return json.loads(function_call)
+    else:
+        return None
+
+
+def chatbot(query):
+    messages = [
+        {"role": "system", "content": "You are a movie search assistant bot who uses TMDB to help users find movies. Think step by step and identify the sequence of function calls that will help to answer."},
+        {"role": "user", "content": query},
+    ]
+
+    tokenized_chat = tokenizer.apply_chat_template(
+        messages, tools=tools, add_generation_prompt=True, tokenize=True, return_tensors="pt")
+
+    outputs = model.generate(tokenized_chat, max_new_tokens=128)
+    answer = tokenizer.batch_decode(outputs[:, tokenized_chat.shape[1]:])[0]
+    tool_call = extract_function_call(answer)
+
+    if tool_call:
+        function_name = tool_call['name']
+        function_params = tool_call['parameters']
+        print("\nfunction_name: ", function_name,
+              "\nfunction_params: ", function_params)
+        function_result = names_to_functions[function_name](**function_params)
+        print(function_result['results'])
+        return function_result['results']
+    else:
+        print("No tool calls found in the answer.")

app.py

@@ -0,0 +1,99 @@
+from flask import Flask, render_template, request, jsonify
+from agent import chatbot
+from functions import query_tmdb
+
+app = Flask(__name__)
+
+
+@app.route('/')
+def index():
+    page = request.args.get('page', 1)
+    url = f'https://api.themoviedb.org/3/movie/popular?page={page}'
+    data = query_tmdb(url)
+    movies = data.get('results', [])
+    total_pages = data.get('total_pages', 1)
+    return render_template('index.html', movies=movies, page=int(page), total_pages=total_pages, title="Popular Movies")
+
+
+@app.route('/movie/<int:movie_id>')
+def movie_details(movie_id):
+    url = f'https://api.themoviedb.org/3/movie/{movie_id}'
+    movie = query_tmdb(url)
+    return jsonify(movie)
+
+
+@app.route('/movie/<int:movie_id>/credits')
+def movie_credits(movie_id):
+    url = f'https://api.themoviedb.org/3/movie/{movie_id}/credits'
+    credits = query_tmdb(url)
+    return jsonify(credits)
+
+
+@app.route('/genre/<int:genre_id>')
+def genre_filter(genre_id):
+    page = request.args.get('page', 1)
+    url = f'https://api.themoviedb.org/3/discover/movie?with_genres={genre_id}&page={page}'
+    data = query_tmdb(url)
+    genre_url = 'https://api.themoviedb.org/3/genre/movie/list'
+    genres = query_tmdb(genre_url).get('genres', [])
+    genre_name = next(
+        (genre['name'] for genre in genres if genre['id'] == genre_id), 'Unknown')
+    return render_template('index.html', movies=data.get('results', []), page=int(page), total_pages=data.get('total_pages', 1), title=f"Movies in Genre: {genre_name}")
+
+
+@app.route('/genres')
+def genres():
+    genre_url = 'https://api.themoviedb.org/3/genre/movie/list'
+    genres = query_tmdb(genre_url).get('genres', [])
+    return render_template('genres.html', genres=genres)
+
+
+@app.route('/cast')
+def cast():
+    page = request.args.get('page', 1)
+    url = f'https://api.themoviedb.org/3/trending/person/week?page={page}'
+    data = query_tmdb(url)
+    persons = data.get('results', [])
+    total_pages = data.get('total_pages', 1)
+    return render_template('cast.html', persons=persons, page=int(page), total_pages=total_pages, title="Trending Cast Members")
+
+
+class PersonInfo:
+    def __init__(self, name, image, bio):
+        self.name = name
+        self.image = image
+        self.bio = bio
+
+
+@app.route('/cast/<int:cast_id>')
+def cast_filter(cast_id):
+    page = request.args.get('page', 1)
+    url = f'https://api.themoviedb.org/3/discover/movie?with_cast={cast_id}&page={page}'
+    data = query_tmdb(url)
+    cast_url = f'https://api.themoviedb.org/3/person/{cast_id}'
+    cast_data = query_tmdb(cast_url)
+    cast_name = cast_data.get('name', 'Unknown')
+    cast_bio = cast_data.get('biography', 'No biography available')
+    cast_image = f"https://image.tmdb.org/t/p/w500{cast_data.get('profile_path', '')}"
+    cast_info = PersonInfo(name=cast_name, image=cast_image, bio=cast_bio)
+    return render_template('index.html',
+                           movies=data.get('results', []),
+                           page=int(page),
+                           total_pages=data.get('total_pages', 1),
+                           title=f"Movies with Actor/Director: {cast_name}",
+                           cast_info=cast_info)
+
+
+@app.route('/query', methods=['GET', 'POST'])
+def query():
+    if request.method == 'POST':
+        query = request.form.get('query')
+        result = chatbot(query)  # Call the chatbot function from agent.py
+        print("Got result: ", result)
+        print("Movie ids are: ", [r['id'] for r in result])
+        return render_template('query.html', result=result)
+    return render_template('query.html')
+
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=5000)

functions.py

@@ -0,0 +1,68 @@
+import requests
+import os
+
+BASE_URL = "https://api.themoviedb.org/3"
+API_KEY = os.environ['TMDB_API_KEY']
+
+
+def query_tmdb(endpoint, params={}):
+    headers = {
+        "accept": "application/json",
+        "Authorization": f"Bearer {API_KEY}"
+    }
+    response = requests.get(endpoint, headers=headers, params=params)
+    return response.json()
+
+
+def discover_movie(include_adult=False, include_video=False, language="en-US", page=1, sort_by="popularity.desc", **kwargs):
+    endpoint = f"{BASE_URL}/discover/movie"
+    params = {}
+
+    for key, value in kwargs.items():
+        if value is not None:
+            params[key] = value
+    response = query_tmdb(endpoint, params=params)
+    return response
+
+
+def get_movie_details(movie_id, append_to_response=None):
+    endpoint = f"{BASE_URL}/movie/{movie_id}"
+    params = {}
+
+    response = query_tmdb(endpoint, params=params)
+    return response
+
+
+def search_person(query, include_adult=False, language="en-US", page=1):
+    endpoint = f"{BASE_URL}/search/person"
+    params = {
+        'query': query,
+        'include_adult': include_adult,
+        'language': language,
+        'page': page
+    }
+
+    response = query_tmdb(endpoint, params=params)
+    return response
+
+
+def get_person_details(person_id, language="en-US", append_to_response=None):
+    endpoint = f"{BASE_URL}/person/{person_id}"
+    params = {
+        'language': language
+    }
+    if append_to_response:
+        params['append_to_response'] = append_to_response
+
+    response = query_tmdb(endpoint, params=params)
+    return response
+
+
+def get_movie_genres(language="en-US"):
+    endpoint = f"{BASE_URL}/genre/movie/list"
+    params = {
+        'language': language
+    }
+
+    response = query_tmdb(endpoint, params=params)
+    return response

requirements.txt

@@ -0,0 +1,6 @@
+Flask
+requests
+transformers
+torch
+bitsandbytes
+accelerate

static/css/style.css

@@ -0,0 +1,94 @@
+#sidebar {
+    width: 200px;
+    position: fixed;
+    height: 100%;
+    left: 0;
+    background-color: #f8f9fa;
+    padding: 20px;
+}
+
+#sidebar ul {
+    list-style-type: none;
+    padding: 0;
+}
+
+#sidebar li {
+    margin-bottom: 10px;
+}
+
+#sidebar a {
+    text-decoration: none;
+    color: #333;
+}
+
+#main-content {
+    margin-left: 200px;
+    padding: 40px;
+}
+
+.tile-element {
+    display: inline-block;
+    width: 200px;
+    margin: 10px;
+    cursor: pointer;
+    vertical-align: top;
+}
+
+.tile-element img {
+    width: 100%;
+}
+
+#side-panel {
+    position: fixed;
+    top: 50%;
+    left: 50%;
+    transform: translate(-50%, -50%);
+    width: 60%;
+    background: white;
+    border: 1px solid #ccc;
+    padding: 20px;
+    z-index: 1000;
+    display: none;
+    flex-direction: row;
+    justify-content: space-between;
+    align-items: flex-start;
+}
+
+#side-panel.active {
+    display: flex;
+}
+
+#close-button {
+    align-self: flex-start;
+    cursor: pointer;
+    font-size: 20px;
+}
+
+#movie-details {
+    padding-left: 20px;
+    text-align: left;
+    flex-direction: column;
+}
+
+#movie-poster {
+    width: 30%;
+    object-fit: contain;
+}
+
+#overlay {
+    display: none;
+    /* Hidden by default */
+    position: fixed;
+    top: 0;
+    left: 0;
+    width: 100%;
+    height: 100%;
+    background: rgba(0, 0, 0, 0.5);
+    /* Semi-transparent background */
+    z-index: 999;
+}
+
+#overlay.active {
+    display: block;
+    /* Show overlay when active */
+}

static/js/script.js

@@ -0,0 +1,62 @@
+
+function generatePanelContent(data) {
+    const { movie, credits } = data;
+
+    const posterUrl = movie.poster_path
+        ? `https://image.tmdb.org/t/p/w200${movie.poster_path}`
+        : 'default_image_url';
+
+    const genres = movie.genres.map(genre => {
+        return `<a href="/genre/${genre.id}">${genre.name}</a>`;
+    }).join(', ');
+
+    const director = credits.crew.find(crewMember => crewMember.job === 'Director');
+    const directorLink = director
+        ? `<a href="/cast/${director.id}">${director.name}</a>`
+        : 'N/A';
+
+    const cast = credits.cast.slice(0, 5).map(castMember => {
+        return `<a href="/cast/${castMember.id}">${castMember.name}</a>`;
+    }).join(', ');
+
+    const movieDetailsInnerHTML = `
+        <h2>${movie.title}</h2>
+        <p>${movie.overview}</p>
+        <strong>Genres:</strong> ${genres}<br>
+        <strong>Director:</strong> ${directorLink}<br>
+        <strong>Cast:</strong> ${cast}`;
+
+    return { posterUrl, movieDetailsInnerHTML };
+}
+
+function showDetails(movieId) {
+    fetch(`/movie/${movieId}`)
+        .then(response => response.json())
+        .then(movie => {
+            fetch(`/movie/${movieId}/credits`)
+                .then(response => response.json())
+                .then(credits => {
+                    const panel = document.getElementById('side-panel');
+                    const overlay = document.getElementById('overlay');
+                    const moviePoster = document.getElementById('movie-poster');
+                    const movieDetails = document.getElementById('movie-details');
+
+                    // Retrieve posterUrl and movieDetailsInnerHTML from generatePanelContent
+                    const { posterUrl, movieDetailsInnerHTML } = generatePanelContent({ movie, credits });
+
+                    // Update the respective DOM elements
+                    moviePoster.src = posterUrl;
+                    movieDetails.innerHTML = movieDetailsInnerHTML;
+
+                    panel.classList.add('active'); // Add active class to slide in
+                    overlay.classList.add('active'); // Show overlay
+                });
+        });
+}
+
+function closePanel() {
+    const panel = document.getElementById('side-panel');
+    const overlay = document.getElementById('overlay');
+    panel.classList.remove('active'); // Remove active class to slide out
+    overlay.classList.remove('active'); // Hide overlay
+}

templates/base.html

@@ -0,0 +1,27 @@
+<!DOCTYPE html>
+<html lang="en">
+
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>TMDB Movies Explorer</title>
+
+    <link rel="stylesheet" href="{{ url_for('static', filename='css/style.css') }}">
+    <script src="{{ url_for('static', filename='js/script.js') }}"></script>
+</head>
+
+<body>
+    <div id="sidebar">
+        <ul>
+            <li><a href="/">Popular Movies</a></li>
+            <li><a href="/genres">By Genre</a></li>
+            <li><a href="/cast">By Cast</a></li>
+            <li><a href="/query">Query</a></li>
+        </ul>
+    </div>
+    <div id="main-content">
+        {% block content %}{% endblock %}
+    </div>
+</body>
+
+</html>

templates/cast.html

@@ -0,0 +1,12 @@
+{% extends 'base.html' %}
+
+{% block content %}
+{% for person in persons %}
+<div class="tile-element">
+    <a href="cast/{{ person.id }}">
+        <img src="https://image.tmdb.org/t/p/w500{{ person.profile_path }}" alt="{{ person.name}}">
+    </a>
+    <h2>{{ person.name }}</h2>
+</div>
+{% endfor %}
+{% endblock %}

templates/genres.html

@@ -0,0 +1,11 @@
+<!-- genres.html -->
+{% extends 'base.html' %}
+
+{% block content %}
+<h1>Genres</h1>
+<ul>
+    {% for genre in genres %}
+    <li><a href="/genre/{{ genre.id }}">{{ genre.name }}</a></li>
+    {% endfor %}
+</ul>
+{% endblock %}

templates/index.html

@@ -0,0 +1,45 @@
+{% extends 'base.html' %}
+
+{% block content %}
+
+{% if cast_info %}
+<div style="display: flex;">
+    <div style="max-width: 30%;">
+        <img src="{{ cast_info.image }}" alt="{{ title }}" style="width: 300px; object-fit: contain;">
+    </div>
+    <div style="max-width: 70%; padding-left: 20px;">
+        <h2>{{ cast_info.name }}</h2>
+        <p>{{ cast_info.bio }}</p>
+    </div>
+</div>
+<h2>{{ title }}</h2>
+{% else %}
+<h1>{{ title }}</h1>
+{% endif %}
+
+<div>
+    {% for movie in movies %}
+    <div class="tile-element" onclick="showDetails({{ movie.id }})">
+        <img src="https://image.tmdb.org/t/p/w500{{ movie.poster_path }}" alt="{{ movie.title }}">
+        <h4>{{ movie.title }}</h4>
+    </div>
+    {% endfor %}
+</div>
+<div>
+    {% if page > 1 %}
+    <a href="?page={{ page - 1 }}">Previous</a>
+    {% endif %}
+    {% if page < total_pages %} <a href="?page={{ page + 1 }}">Next</a>
+        {% endif %}
+</div>
+
+<div id="overlay" onclick="closePanel()"></div> <!-- Overlay for closing panel -->
+<div id="side-panel">
+    <img id="movie-poster" src="" alt="${movie.title} Poster" style="max-width: 150px;">
+    <div id="movie-details">
+        <!-- Movie details will be injected here -->
+    </div>
+    <span id="close-button" onclick="closePanel()">&times;</span>
+
+</div>
+{% endblock %}

templates/query.html

@@ -0,0 +1,23 @@
+{% extends 'base.html' %}
+
+{% block content %}
+<h1>Query</h1>
+<form action="/query" method="post">
+    <input type="text" name="query" placeholder="Enter your query...">
+    <input type="submit" value="Submit">
+</form>
+
+{% if result %}
+<h2>Query Result</h2>
+<div>
+    {% for movie in result %}
+    <div class="tile-element" onclick="showDetails({{ movie.id }})">
+        <img src="https://image.tmdb.org/t/p/w500{{ movie.poster_path }}" alt="{{ movie.title }}">
+        <h4>{{ movie.title }}</h4>
+    </div>
+    {% endfor %}
+</div>
+{% elif result is defined %}
+<p>No results found.</p>
+{% endif %}
+{% endblock %}

tools.py

@@ -0,0 +1,251 @@
+tools = [
+    {
+        "type": "function",
+        "function": {
+            "name": "discover_movie",
+            "description": "Find movies using over 30 filters and sort options",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "region": {
+                        "type": "string",
+                        "description": "ISO 3166-1 code to filter release dates",
+                    },
+                    "sort_by": {
+                        "type": "string",
+                        "description": "Sort the results",
+                    },
+                    "certification_country": {
+                        "type": "string",
+                        "description": "Used in conjunction with the certification filter",
+                    },
+                    "certification": {
+                        "type": "string",
+                        "description": "Filter results with a valid certification from the certification_country",
+                    },
+                    "certification.lte": {
+                        "type": "string",
+                        "description": "Filter and only include movies that have a certification that is less than or equal to the specified value",
+                    },
+                    "certification.gte": {
+                        "type": "string",
+                        "description": "Filter and only include movies that have a certification that is greater than or equal to the specified value",
+                    },
+                    "include_adult": {
+                        "type": "boolean",
+                        "description": "Choose whether to include adult (pornography) content in the results",
+                    },
+                    "include_video": {
+                        "type": "boolean",
+                        "description": "Choose whether to include videos in the results",
+                    },
+                    "page": {
+                        "type": "integer",
+                        "description": "Specify which page to query",
+                    },
+                    "primary_release_year": {
+                        "type": "integer",
+                        "description": "Filter the results to only include movies that have a primary release year that equals the specified value",
+                    },
+                    "primary_release_date.gte": {
+                        "type": "string",
+                        "description": "Filter and only include movies that have a primary release date that is greater or equal to the specified value",
+                    },
+                    "primary_release_date.lte": {
+                        "type": "string",
+                        "description": "Filter and only include movies that have a primary release date that is less than or equal to the specified value",
+                    },
+                    "release_date.gte": {
+                        "type": "string",
+                        "description": "Filter and only include movies that have a release date (looking at all release dates) that is greater or equal to the specified value",
+                    },
+                    "release_date.lte": {
+                        "type": "string",
+                        "description": "Filter and only include movies that have a release date (looking at all release dates) that is less than or equal to the specified value",
+                    },
+                    "with_release_type": {
+                        "type": "integer",
+                        "description": "Specify a comma (AND) or pipe (OR) separated value to filter release types",
+                    },
+                    "year": {
+                        "type": "integer",
+                        "description": "Filter the results to only include movies that have a release year that equals the specified value",
+                    },
+                    "vote_count.gte": {
+                        "type": "integer",
+                        "description": "Filter and only include movies that have a vote count that is greater or equal to the specified value",
+                    },
+                    "vote_count.lte": {
+                        "type": "integer",
+                        "description": "Filter and only include movies that have a vote count that is less than or equal to the specified value",
+                    },
+                    "vote_average.gte": {
+                        "type": "number",
+                        "description": "Filter and only include movies that have a rating that is greater or equal to the specified value",
+                    },
+                    "vote_average.lte": {
+                        "type": "number",
+                        "description": "Filter and only include movies that have a rating that is less than or equal to the specified value",
+                    },
+                    "with_cast": {
+                        "type": "string",
+                        "description": "A comma separated list of person ID's to filter the results with",
+                    },
+                    "with_crew": {
+                        "type": "string",
+                        "description": "A comma separated list of person ID's to filter the results with",
+                    },
+                    "with_people": {
+                        "type": "string",
+                        "description": "A comma separated list of person ID's to filter the results with",
+                    },
+                    "with_companies": {
+                        "type": "string",
+                        "description": "A comma separated list of production company ID's to filter the results with",
+                    },
+                    "with_genres": {
+                        "type": "string",
+                        "description": "A comma separated list of genre ID's to filter the results with",
+                    },
+                    "without_genres": {
+                        "type": "string",
+                        "description": "A comma separated list of genre ID's to exclude from the results",
+                    },
+                    "with_keywords": {
+                        "type": "string",
+                        "description": "A comma separated list of keyword ID's to filter the results with",
+                    },
+                    "without_keywords": {
+                        "type": "string",
+                        "description": "A comma separated list of keyword ID's to exclude from the results",
+                    },
+                    "with_runtime.gte": {
+                        "type": "integer",
+                        "description": "Filter and only include movies that have a runtime that is greater or equal to the specified value",
+                    },
+                    "with_runtime.lte": {
+                        "type": "integer",
+                        "description": "Filter and only include movies that have a runtime that is less than or equal to the specified value",
+                    },
+                    "with_original_language": {
+                        "type": "string",
+                        "description": "Specify an ISO 639-1 string to filter results by their original language value",
+                    },
+                    "with_watch_providers": {
+                        "type": "string",
+                        "description": "A comma or pipe separated list of watch provider ID's to filter the results with",
+                    },
+                    "watch_region": {
+                        "type": "string",
+                        "description": "An ISO 3166-1 code to filter the watch provider results",
+                    },
+                    "with_watch_monetization_types": {
+                        "type": "string",
+                        "description": "Filter the results by monetization type",
+                    },
+                    "without_companies": {
+                        "type": "string",
+                        "description": "A comma separated list of production company ID's to filter the results with",
+                    },
+                },
+                "required": [],
+            },
+        },
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "get_movie_details",
+            "description": "Get the top level details of a movie by ID",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "movie_id": {
+                        "type": "integer",
+                        "description": "The ID of the movie to get details for",
+                    },
+                    "append_to_response": {
+                        "type": "string",
+                        "description": "Comma-separated list of sub requests to append to the response",
+                    },
+                },
+                "required": ["movie_id"],
+            },
+        },
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "search_person",
+            "description": "Search for people in the entertainment industry.",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "query": {
+                        "type": "string",
+                        "description": "The search query for the person"
+                    },
+                    "include_adult": {
+                        "type": "boolean",
+                        "description": "Include adult (pornography) content in the results",
+                        "default": False
+                    },
+                    "language": {
+                        "type": "string",
+                        "description": "Language for the search results",
+                        "default": "en-US"
+                    },
+                    "page": {
+                        "type": "integer",
+                        "description": "Page number of results",
+                        "default": 1
+                    }
+                },
+                "required": ["query"]
+            }
+        }
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "get_person_details",
+            "description": "Get detailed information about a specific person.",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "person_id": {
+                        "type": "integer",
+                        "description": "The ID of the person to get details for"
+                    },
+                    "language": {
+                        "type": "string",
+                        "description": "Language for the person details",
+                        "default": "en-US"
+                    },
+                    "append_to_response": {
+                        "type": "string",
+                        "description": "Comma-separated list of additional details to append to the response (e.g., 'images,credits')"
+                    }
+                },
+                "required": ["person_id"]
+            }
+        }
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "get_movie_genres",
+            "description": "Get the list of official genres for movies.",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "language": {
+                        "type": "string",
+                        "description": "Language for the genre names",
+                        "default": "en-US"
+                    }
+                }
+            }
+        }
+    }
+]