alexkueck commited on
Commit
67a7465
1 Parent(s): 4806881

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +21 -12
app.py CHANGED
@@ -153,11 +153,23 @@ def add_file(history, file, prompt):
153
  def file_anzeigen(file):
154
  return gr.File(visible=True), file.name
155
 
156
- def create_picture_backup(history, prompt):
157
- client = OpenAI()
158
- response = client.images.generate(model="dall-e-3", prompt=prompt,size="1024x1024",quality="standard",n=1,)
159
- image_url = response.data[0].url
160
- return image_url
 
 
 
 
 
 
 
 
 
 
 
 
161
 
162
  def transfer_input(inputs):
163
  textbox = reset_textbox()
@@ -378,10 +390,7 @@ def create_picture(history, prompt):
378
  # using the Image module from PIL library to view the image
379
  #Image.open(response.raw)
380
 
381
- def umwandeln_fuer_anzeige(image):
382
- buffer = io.BytesIO()
383
- image.save(buffer, format='PNG')
384
- return buffer.getvalue()
385
 
386
  def generate_auswahl(prompt, file, chatbot, history, rag_option, model_option, openai_api_key, k=3, top_p=0.6, temperature=0.5, max_new_tokens=4048, max_context_length_tokens=2048, repetition_penalty=1.3,):
387
  if (prompt.find('zeichnen') != -1):
@@ -449,9 +458,9 @@ def generate_text (prompt, file, chatbot, history, rag_option, model_option, ope
449
  else:
450
  print("file bild uplad....................")
451
  print(file)
452
- image = Image.open(io.BytesIO(file_anzeigen))
453
- image_64 = umwandeln_fuer_anzeige(image)
454
- prompt_neu = prompt + b64encode(image_64).decode("utf-8").path()
455
  print("prompt_neu............................")
456
  print(prompt_neu)
457
  history_text_und_prompt = generate_prompt_with_history_openai(prompt_neu, history)
 
153
  def file_anzeigen(file):
154
  return gr.File(visible=True), file.name
155
 
156
+ def umwandeln_fuer_anzeige(image):
157
+ buffer = io.BytesIO()
158
+ image.save(buffer, format='PNG')
159
+ return buffer.getvalue()
160
+
161
+ def process_image(image_path, prompt):
162
+ # Convert image to base64
163
+ with open(image_path, "rb") as image_file:
164
+ encoded_string = base64.b64encode(image_file.read()).decode()
165
+
166
+ # Prepare the data for the API request (specific to the API you're using)
167
+ data = {
168
+ 'image': encoded_string,
169
+ 'input': prompt,
170
+ }
171
+ return data
172
+
173
 
174
  def transfer_input(inputs):
175
  textbox = reset_textbox()
 
390
  # using the Image module from PIL library to view the image
391
  #Image.open(response.raw)
392
 
393
+
 
 
 
394
 
395
  def generate_auswahl(prompt, file, chatbot, history, rag_option, model_option, openai_api_key, k=3, top_p=0.6, temperature=0.5, max_new_tokens=4048, max_context_length_tokens=2048, repetition_penalty=1.3,):
396
  if (prompt.find('zeichnen') != -1):
 
458
  else:
459
  print("file bild uplad....................")
460
  print(file)
461
+ #image = Image.open(io.BytesIO(file_anzeigen))
462
+ #image_64 = umwandeln_fuer_anzeige(image)
463
+ prompt_neu = process_image(file, prompt)
464
  print("prompt_neu............................")
465
  print(prompt_neu)
466
  history_text_und_prompt = generate_prompt_with_history_openai(prompt_neu, history)