Welcome to the third lesson in our course on building an image generation web application with Django! In our previous lessons, we've built a solid foundation for our application by creating the HTML structure and styling it with CSS. We now have a visually appealing interface with tabs, form elements, and containers for displaying generated images.
However, our application is still static — clicking buttons doesn't do anything, and there's no way to actually generate images or switch between tabs. This is where JavaScript comes in. JavaScript is the programming language that brings web pages to life by adding interactivity and dynamic behavior.
In this lesson, we'll focus on implementing the client-side functionality that will allow users to interact with our application. We'll write JavaScript code to handle tab navigation, capture user input, send requests to our backend API, and display the generated images.
The JavaScript file we'll be creating will serve as the bridge between our user interface and the backend Django application. It will handle all the client-side logic, from validating user input to making API calls and updating the DOM based on the responses.
Let's get started by implementing the tab navigation functionality, which will allow users to switch between the "Generate Image" and "View History" tabs.
Our application has two main tabs: "Generate Image" and "View History." We need to implement a function that will show the selected tab and hide the others when a user clicks on a tab button.
Let's create a file named script.js
in the static
folder of our Django application. This is the same folder where we placed our style.css
file in the previous lesson.
Now, let's implement the openTab()
function that will handle the tab switching:
This function does two main things:
- First, it selects all elements with the class
tab-content
usingdocument.querySelectorAll()
and hides them by setting theirdisplay
style property to'none'
. - Then, it selects the specific tab content element with the ID matching the
tabName
parameter usingdocument.getElementById()
and shows it by setting itsdisplay
style property to'block'
.
When a user clicks on a tab button, this function will be called with the ID of the corresponding tab content as the argument. For example, when the user clicks on the "Generate Image" tab button, the function will be called as openTab('generate')
, and when they click on the "View History" tab button, it will be called as openTab('history')
.
Remember that in our HTML, we've already set up the tab buttons to call this function with the appropriate arguments:
With this function in place, users can now switch between tabs by clicking on the tab buttons. The selected tab's content will be displayed, while the other tab's content will be hidden.
This approach is also backward compatible and scalable. If you decide to add more .tab-content
divs in the future (for example, to introduce new tabs), you won't need to change the openTab()
function. It will automatically hide all tab content sections and only display the one corresponding to the selected tab. This makes it easy to extend your application with additional tabs without modifying your JavaScript logic.
Now that we have the tab navigation working, let's move on to implementing the image generation functionality.
The core functionality of our application is generating images based on user input. We need to implement a function that will capture the user's input, validate it, and send it to our backend API.
Let's add the generateImage()
function to our script.js
file:
Let's break down this function step by step.
First, we need to retrieve the values from the input fields where the user enters a prompt and selects an aspect ratio:
These values will be included in the request to our backend API.
Next, we get references to the key elements in the UI that we'll need to interact with:
These elements will be used to show a loading state, hide the button, and display the generated image.
Before proceeding, we check that the user has entered a prompt:
This prevents the user from submitting an empty prompt and ensures meaningful input is sent to the backend.
To indicate that the image is being generated, we update the UI accordingly:
This gives the user feedback and prepares the page to display a new image.
To send the user's input to our backend API, we'll use the Fetch API, which is a modern interface for making HTTP requests in JavaScript.
Let's break down this fetch request:
- The first argument is the URL of the endpoint we're sending the request to:
/api/generate_image/
. - The second argument is an object with options for the request:
method: 'POST'
specifies that we're sending a POST request, which is appropriate for creating new resources (in this case, generating a new image).headers: { 'Content-Type': 'application/json' }
tells the server that we're sending JSON data.body: JSON.stringify({ user_input: promptInput, aspect_ratio: aspectRatio })
converts our JavaScript object containing the user's input into a JSON string, which is the format our server expects.
The fetch()
function returns a Promise that resolves to the Response object representing the response to the request. Once we receive a response from the backend, we can use the .then()
method to parse the JSON and update the UI:
This code resets the UI, checks for errors, and displays the generated image if the request was successful.
The first line takes the Response object and calls its .json()
method, which returns another Promise that resolves to the parsed JSON data from the response body.
We can chain another .then()
to handle the parsed data:
If there was an error, we alert it.
If everything is ok, we will enter the else block where we:
- Create a new
img
element usingdocument.createElement('img')
. - Set its
src
attribute to a data URL that includes the base64-encoded image data from the response. The formatdata:image/png;base64,${data.image}
tells the browser that this is a PNG image encoded in base64. - Append the image element to the
imageContainer
usingimageContainer.appendChild(img)
.
This will display the generated image on the page, allowing the user to see the result of their prompt.
It's worth noting that we're using a data URL to display the image directly, rather than loading it from a separate URL. This is a common approach for small images or when you want to avoid making additional HTTP requests.
Finally, we handle any unexpected issues that may occur during the request:
This ensures the user is informed and the UI returns to a usable state if something goes wrong.
With this code in place, our application can now generate images based on user input and display them on the page. The user can enter a prompt, select an aspect ratio and style, click the "Generate Image" button, and see the resulting image appear below the form.
Congratulations! In this lesson, we've implemented the client-side functionality for our image generation web application. Let's review what we've accomplished:
- We created a JavaScript file to handle the client-side logic of our application.
- We implemented the
openTab()
function to allow users to switch between tabs. - We built the
generateImage()
function to capture user input, validate it, and send it to our backend API. - We used the Fetch API to make asynchronous requests to our server.
- We processed the API responses and updated the UI to display the generated images.
Our application now has a functional user interface that allows users to generate images based on their input. However, there's still room for improvement. In the next lesson, we'll enhance our application by adding more robust loading states and error handling, as well as implementing the functionality to view previously generated images in the history tab.
In the practice exercises that follow this lesson, you'll have the opportunity to experiment with the JavaScript code we've written. You might try modifying the validation logic, adding additional features, or improving the error handling. These exercises will help reinforce your understanding of the concepts we've covered and give you hands-on experience with client-side JavaScript development.
Remember, JavaScript is a powerful language that can greatly enhance the user experience of your web applications. The skills you're learning in this course will be valuable not only for this specific project but for many other web development tasks you might encounter in the future.
