crude version, but working

.vscode/settings.json

@@ -0,0 +1,5 @@
+{
+    "editor.formatOnPaste": true,
+    "editor.detectIndentation": false,
+    "python.formatting.provider": "black"
+}

app.py

@@ -1,18 +1,121 @@
 import gradio as gr
 import openai
 import os
+import IPython
 import random
-
+import requests
+import json
+import time
+import math
+import io
+
+from stability_sdk import client as sclient
+from io import BytesIO
 from dotenv import load_dotenv
 from openai import OpenAI
+from PIL import Image
+import stability_sdk.interfaces.gooseai.generation.generation_pb2 as generation
 
 load_dotenv()
 
+sdurl = "https://stablediffusionapi.com/api/v3/text2img"
 openai.api_key = os.getenv('OPENAI_API_KEY')
-print(openai.api_key)
-print (os.environ.get("OPENAI_API_KEY"))
+#print(openai.api_key)
+#print (os.environ.get("OPENAI_API_KEY"))
 client = openai.OpenAI(organization='org-eiNl8e4nk93VLQFDb4EBz9JG')
 #openai.api_key = os.environ['openai']  #for huggingface i think
+os.environ['STABILITY_HOST'] = 'grpc.stability.ai:443'
+stability_api = sclient.StabilityInference(
+    key=os.getenv('STABILITY_KEY'), 
+    verbose=True,
+	engine="stable-diffusion-xl-1024-v1-0"
+)
+
+
+
+#generates an AI picture of your char
+def draw(thechar):
+    thechar=thechar.split("$$$")[0].strip()
+    moretime=50
+    cfg=8.0
+    noinit=0
+    r2="1"
+    r3="1"
+    filename="drunkgrogu"+str(r2)+".png"
+    filenamefilter="calvin"+str(r3)+".png"
+    #await ctx.channel.send(file=discord.File(filename))
+    #stable-diffusion-xl-1024-v0-9
+    notoldguy="bad anatomy, fuzzy, out of focus, cut off, low res"
+
+    try :
+        if noinit :
+            print("initimg")
+            answers = stability_api.generate(
+				prompt=[generation.Prompt(text=thechar,parameters=generation.PromptParameters(weight=1)), 
+				generation.Prompt(text=notoldguy,parameters=generation.PromptParameters(weight=-1))], 				
+				steps=moretime, # defaults to 50 if not specified
+			)
+        else :
+            print("noinitimg:"+thechar+"\nneg:"+notoldguy)
+            answers = stability_api.generate(
+					prompt=[generation.Prompt(text=thechar,parameters=generation.PromptParameters(weight=1)), 
+					generation.Prompt(text=notoldguy,parameters=generation.PromptParameters(weight=-1))], 
+					cfg_scale=cfg,				
+					steps=moretime, # defaults to 50 if not specified
+					width=1024,
+					height=1024
+				)
+    except Exception as eeeee :
+        print(eeeee)
+        
+
+    try :
+		# iterating over the generator produces the api response
+        for resp in answers:
+			#print(resp)
+            for artifact in resp.artifacts:
+				#print("fin rease:"+str(artifact.finish_reason))
+                if artifact.finish_reason == generation.FILTER:
+                    if random.randrange(0,2)==1 :
+                        print("await ctx.channel.send(file=discord.File(filenamefilter))")				
+                    else :
+                       print(" await ctx.channel.send(file=discord.File(filename))")
+#warnings.warn("Your request activated the API's safety filters and could not be processed.   Please modify the prompt and try again.")
+                if artifact.type == generation.ARTIFACT_IMAGE:
+					#print("we got an image")
+                    img = Image.open(io.BytesIO(artifact.binary))
+                    #oldoldguy=oldoldguy.replace("|","--")
+                  #  oldoldguy = oldoldguy.replace("*","")
+					#metadata = PngInfo()
+					#metadata.add_text("prompt", oldguy)
+                    thechar=thechar.replace("Character Description","")
+                    filename="dndChar"+str(math.trunc(time.time()))+thechar[:80].replace("\n","").replace(",","")
+			
+                    filename=filename[:245]+".png"
+                    img.save(filename)
+
+					##filename="image5r"+str(math.trunc(time.time()))+".png"
+					#print(str(img))
+					##filename=filename[:245]+".png"
+					#print(filename)
+					##img.save(filename)
+					#filename="image5r"+str(math.trunc(time.time()))+".png"
+					#img.save(filename)
+					#display(img)
+
+    except Exception as eeee :
+
+        print(eeee.details)
+        print("Error. Have a random image instead:");		
+
+        
+    #print(response.text)
+    #print(response.output)
+    print(filename)
+    imageBox = gr.Image(value=Image.open(filename))
+    imageBox2 = img
+    return img
+
 
 #generates an AI description of your character
 def describe(names,wis,char,str,int,dex,con):
@@ -43,21 +146,19 @@ def describe(names,wis,char,str,int,dex,con):
     if not result :
         result = "Could you be any more boring?"
         
+    #draw(result.split("$$$")[0],outputs=imageBox2)
+    return result#result.replace("$$$","")
 
-    return result
-
-#generates an AI description of your character
+#generates an AI nam,
 def makeName():
 
-    prom=f"We're generating Dungeons and Dragons characters for a new campaign, and we need a name." 
-    prom+=f" Please imagine and give me just a character name for the player to use."
+    prom=f"We're generating Dungeons and Dragons characters for a new campaign, and we need a good, original name." 
+    prom+=f" Please imagine and give me just a character name for the player to use. Be creative and fun. No duds."
     completion = client.completions.create(
 		model='gpt-3.5-turbo-instruct',
 		prompt=prom,
 		max_tokens=50,
-		temperature=0.77,
-		frequency_penalty=0.2,
-		presence_penalty= 0.25)
+		temperature=0.97)
 
     result =completion.choices[0].text
 	#print(dict(completion).get('usage'))
@@ -69,7 +170,7 @@ def makeName():
         result = "The Doctor"
         
 
-    return result
+    return result.strip()
 
 def stat( ndice, dicerank):
    # return ndice + dicerank
@@ -101,13 +202,14 @@ with gr.Blocks() as statBlock:
 
     rollButt = gr.Button(value="Roll Stats")  
     describeButt = gr.Button(value="Generate Description")  
-        
+    imgButt = gr.Button(value="Generate Image")       
     outputBox=gr.Textbox(label="The character",show_label=True)
-
+    imageBox = gr.Image()
+    imageBox2 = gr.Image()
     nameButt.click(makeName,inputs=[],outputs=[nameBox])
     rollButt.click(stat, inputs=[dice,gr.Number(value=6, visible=False,precision=0)],outputs=[wisBox,charBox,strBox,intBox,dexBox,conBox])
     describeButt.click(describe, outputs=[outputBox],inputs=[nameBox,wisBox,charBox,strBox,intBox,dexBox,conBox])
+    imgButt.click(draw,inputs=[outputBox],outputs=[imageBox2])
 statBlock.launch()
 
 
-