Update app.py

app.py

@@ -34,46 +34,63 @@ if not check_password():
     st.stop()  # Do not continue if check_password is not True.
 
 
-openai.api_type = "azure"
-openai.api_version = "2023-12-01-preview"
-openai.api_base = "https://tensora-oai-france.openai.azure.com/"
-openai.api_key = os.getenv("france_key")
+swedenclient = AzureOpenAI(
+    api_version = "2023-12-01-preview",
+    azure_endpoint = os.getenv("sweden"),
+    api_key = os.getenv("sweden_key"),
+)
 
 
+usclient = AzureOpenAI(
+    api_version = "2023-12-01-preview",
+    azure_endpoint = os.getenv("us"),
+    api_key = os.getenv("us_key"),
+)
+
+
+prompt = os.getenv("prompt")
+headers = {"Authorization": f"Bearer {os.getenv('imageapi')}"}
+dalle_prompt = os.getenv("dalle_prompt")
+
 prompt = "Please analyze a complete news article and generate a suitable DALL·E prompt in response. The prompt should provide a general overview without being overly specific and the picture should be easy to interpret. For technical purposes, please only provide the DALL·E prompt without any additional information."
 headers = {"Authorization": f"Bearer {os.getenv('imageapi')}"}
 url = "https://api.edenai.run/v2/image/generation"
 
 
 article = st.text_input("Newsarticle to Image")
+pain_points = st.text_input("Pain Points for the picture")
+amount_pictures = st.number_input("Anzahl Bilder", 1 , 2)
 
 if st.button("Get Picture from Text"):
-    try:
-        response = openai.ChatCompletion.create(
-                        engine="gpt-4-0613",
-                        temperature = 0.2,
-                        messages=[
-                            {"role": "system", "content": prompt},
-                            {"role": "system", "content": article}
-                        ],
-                    )
-    except Exception as e:
-        st.error(f"ChatGPT Error {e}")
-    st.write(response.choices[0].message["content"])
-    payload = {
-        "providers": "openai",
-        "settings" : {"openai": "dall-e-3"},
-        "text": response.choices[0].message["content"],
-        "resolution": "1024x1024",
-        "fallback_providers": "openai"
-        }
-    
-    #st.code(payload)
-    
-    response = requests.post(url, json=payload, headers=headers)
-    result = json.loads(response.text)
-    responseimg = requests.get(result['openai']['items'][0]['image_resource_url'])
-    if response.status_code == 200:
-        st.image(responseimg.content)
-    else:
-        st.error("Failed download")
+    for i in range(amount_pictures):
+        try:
+            response = usclient.chat.completions.create(
+                            model="gpt-4-0613",
+                            temperature = 0.2,
+                            messages=[
+                                {"role": "system", "content": prompt + pain_points},
+                                {"role": "system", "content": article}
+                            ],
+                        )
+        except Exception as e:
+            st.error(f"ChatGPT Error {e}")
+        st.code(response.choices[0].message.content)
+
+        try:
+            st.code(response.choices[0].message.content + dalle_prompt)
+            responseimg = swedenclient.images.generate(
+                            model="dalle-3",
+                            prompt =  response.choices[0].message.content + dalle_prompt,
+                            n = 1
+                        )
+        except Exception as e2:
+            st.error(f"Dall-E Error {e2}")
+        
+        image_url = json.loads(responseimg.model_dump_json())['data'][0]['url']
+
+        responseimg2 = requests.get(image_url)
+        if responseimg2.status_code ==200:
+            st.image(responseimg2.content)
+        else:
+            st.error("Failed download")
+