Generating Video Summaries with the OpenAI API in Java

import org.json.JSONArray;
import org.json.JSONObject;

String transcriptionText = "Your transcription text goes here.";

JSONObject systemMessage = new JSONObject();
systemMessage.put("role", "system");
systemMessage.put("content",
    "You are an expert content analyst and summarizer with these capabilities:\n" +
    "- Extracting key points while maintaining context\n" +
    "- Identifying main themes and core messages\n" +
    "- Preserving critical details while reducing length\n" +
    "- Maintaining the original tone and intent\n" +
    "- Organizing information hierarchically\n\n" +
    "Format your summaries with:\n" +
    "1. A one-sentence overview\n" +
    "2. 2-3 key takeaways\n" +
    "3. Important details or quotes (if any)"
);

JSONObject userMessage = new JSONObject();
userMessage.put("role", "user");
userMessage.put("content",
    "Create a structured summary of this transcription. " +
    "Focus on the core message and key points while maintaining " +
    "context and critical details.\n\n" +
    "Transcription:\n" + transcriptionText
);

JSONArray messages = new JSONArray();
messages.put(systemMessage);
messages.put(userMessage);

JSONObject requestBody = new JSONObject();
requestBody.put("model", "gpt-4o");
requestBody.put("messages", messages);

import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;

private static final String OPENAI_API_KEY = System.getenv("OPENAI_API_KEY");

// Set up the HTTP connection
URL url = new URL("https://api.openai.com/v1/chat/completions");
HttpURLConnection conn = (HttpURLConnection) url.openConnection();
conn.setRequestMethod("POST");
conn.setRequestProperty("Authorization", "Bearer " + OPENAI_API_KEY);
conn.setRequestProperty("Content-Type", "application/json");
conn.setDoOutput(true);

// Write the JSON payload to the request body
try (OutputStream os = conn.getOutputStream()) {
    byte[] input = requestBody.toString().getBytes("utf-8");
    os.write(input, 0, input.length);
}

// Read the response
int status = conn.getResponseCode();
InputStream is = (status < HttpURLConnection.HTTP_BAD_REQUEST) ?
    conn.getInputStream() : conn.getErrorStream();

BufferedReader in = new BufferedReader(new InputStreamReader(is, "utf-8"));
StringBuilder response = new StringBuilder();
String line;
while ((line = in.readLine()) != null) {
    response.append(line.trim());
}
in.close();

// Parse the response JSON to extract the summary
JSONObject jsonResponse = new JSONObject(response.toString());
JSONArray choices = jsonResponse.getJSONArray("choices");
if (choices.length() > 0) {
    JSONObject message = choices.getJSONObject(0).getJSONObject("message");
    return message.getString("content");
} else {
    return null;
}

public class MediaProcessorService {

    private static final String OPENAI_API_KEY = System.getenv("OPENAI_API_KEY");

    public String summarizeTranscription(String transcriptionText) {
        try {
            // Step 2: Construct the JSON payload
            JSONObject systemMessage = new JSONObject();
            systemMessage.put("role", "system");
            systemMessage.put("content",
                "You are an expert content analyst and summarizer with these capabilities:\n" +
                "- Extracting key points while maintaining context\n" +
                "- Identifying main themes and core messages\n" +
                "- Preserving critical details while reducing length\n" +
                "- Maintaining the original tone and intent\n" +
                "- Organizing information hierarchically\n\n" +
                "Format your summaries with:\n" +
                "1. A one-sentence overview\n" +
                "2. 2-3 key takeaways\n" +
                "3. Important details or quotes (if any)"
            );

            JSONObject userMessage = new JSONObject();
            userMessage.put("role", "user");
            userMessage.put("content",
                "Create a structured summary of this transcription. " +
                "Focus on the core message and key points while maintaining " +
                "context and critical details.\n\n" +
                "Transcription:\n" + transcriptionText
            );

            JSONArray messages = new JSONArray();
            messages.put(systemMessage);
            messages.put(userMessage);

            JSONObject requestBody = new JSONObject();
            requestBody.put("model", "gpt-4o");
            requestBody.put("messages", messages);

            // Step 3: Set up the HTTP connection
            URL url = new URL("https://api.openai.com/v1/chat/completions");
            HttpURLConnection conn = (HttpURLConnection) url.openConnection();
            conn.setRequestMethod("POST");
            conn.setRequestProperty("Authorization", "Bearer " + OPENAI_API_KEY);
            conn.setRequestProperty("Content-Type", "application/json");
            conn.setDoOutput(true);

            // Step 4: Send request and read response
            try (OutputStream os = conn.getOutputStream()) {
                byte[] input = requestBody.toString().getBytes("utf-8");
                os.write(input, 0, input.length);
            }

            int status = conn.getResponseCode();
            InputStream is = (status < HttpURLConnection.HTTP_BAD_REQUEST) ?
                conn.getInputStream() : conn.getErrorStream();

            BufferedReader in = new BufferedReader(new InputStreamReader(is, "utf-8"));
            StringBuilder response = new StringBuilder();
            String line;
            while ((line = in.readLine()) != null) {
                response.append(line.trim());
            }
            in.close();

            // Step 5: Parse the response
            JSONObject jsonResponse = new JSONObject(response.toString());
            JSONArray choices = jsonResponse.getJSONArray("choices");
            if (choices.length() > 0) {
                JSONObject message = choices.getJSONObject(0).getJSONObject("message");
                return message.getString("content");
            } else {
                return null;
            }
        } catch (Exception e) {
            System.out.println("Error generating summary: " + e.getMessage());
            return null;
        }
    }
}

public class Main {
    public static void main(String[] args) {
        MediaProcessorService service = new MediaProcessorService();
        String transcription = "Your transcription text goes here.";
        String summary = service.summarizeTranscription(transcription);
        System.out.println("Summary:\n" + summary);
    }
}