app.py

import json
import openai
import os
import pandas as pd
from pprint import pprint
import tiktoken
import gradio as gr

openai.api_key = os.getenv("OPENAI_API_KEY")
fine_tuned_model_id = os.getenv("fine_tuned_model_id")

def fine_tune_test(system_message, prompt):
    test_messages = []
    test_messages.append({"role": "system", "content": system_message})
    user_message = prompt
    test_messages.append({"role": "user", "content": user_message})
    
    pprint("==================================================")
    pprint(test_messages)
    response = openai.ChatCompletion.create(
      model=fine_tuned_model_id, messages=test_messages, temperature=0, max_tokens=500
    )
    print(response["choices"][0]["message"]["content"])
    return(response["choices"][0]["message"]["content"])

iface = gr.Interface(fn=fine_tune_test,inputs=[gr.Textbox(label="System Prompt", value="Act as a perfectionist and map the given product by doing critical reasoning."), gr.Textbox(label="User Prompt")],
        outputs=[gr.Textbox(label="Fine Tuned GPT Repsonse")],
        title="GPT 3.5-Turbo Fine Tuning")


iface.launch(share=True)