upgraded GPT to GPT-4

app.py

@@ -8,6 +8,8 @@ from diffusers import StableDiffusionPipeline
 import gradio as gr
 from story_generator import StoryGenerator
 import torch
+import requests 
+import json 
 
 TITLE = '# Fictionista 🧚'
 DESCRIPTION = '''#
@@ -60,7 +62,7 @@ with gr.Blocks(css='style.css') as image_gen_block:
                                       label='Angle')
                     run_button = gr.Button('Run')
                 with gr.Column():
-                    result = gr.Image(label='Result', elem_id='result')
+                    result = gr.Image(label='Result', elem_id='Character')
 
                         # City generation tab
         with gr.TabItem('City'):
@@ -82,9 +84,30 @@ with gr.Blocks(css='style.css') as image_gen_block:
 
 
 
-    def generate_story(prompt, api_key):
-        sg = StoryGenerator(api_key)
-        return sg.generate_story(prompt)
+    def generate_story(prompt, API_KEY, model="gpt-4", temperature=1, max_tokens=None):
+        API_ENDPOINT = "https://api.openai.com/v1/chat/completions"
+
+        headers = {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {API_KEY}",
+        }
+        prompt = [{"role": "user", "content":prompt}]
+        data = {
+            "model": model,
+            "messages": prompt,
+            "temperature": temperature,
+        }
+
+        if max_tokens is not None:
+            data["max_tokens"] = max_tokens
+
+        response = requests.post(API_ENDPOINT, headers=headers, data=json.dumps(data))
+
+        if response.status_code == 200:
+            return response.json()["choices"][0]["message"]["content"]
+        else:
+            raise Exception(f"Error {response.status_code}: {response.text}")
+        
 
     def generate_city():
         city_prompt = f"render 2048K ultra realistic render high definition a very technologically advanced city, cyberpunk, hyper realistic, detailed, ray tracing, POV-Ray, metallic, Sci-Fi"

story_generator.py

@@ -1,20 +0,0 @@
-import openai
-
-class StoryGenerator:
-    def __init__(self, api_key):
-
-        openai.api_key = api_key
-        self.completion_model = "text-davinci-002"
-
-
-    def generate_story(self, prompt):
-        prompt = f"Once upon a time, {prompt}"
-        response = openai.Completion.create(
-            engine=self.completion_model,
-            prompt=prompt,
-            max_tokens=1024,
-            n=1,
-            stop=None,
-            temperature=0.5,
-        )
-        return response.choices[0].text