Update app.py

app.py

@@ -31,25 +31,21 @@ except Exception as e:
 # Set device: CUDA if available, else CPU
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
-model_id1 = "dreamlike-art/dreamlike-diffusion-1.0"
+model_id1 = os.getenv("API_KEY")
+# model_id1 = "dreamlike-art/dreamlike-diffusion-1.0"
 pipe = StableDiffusionPipeline.from_pretrained(model_id1, torch_dtype=torch.float16, use_safetensors=True)
 pipe = pipe.to(device)
 
 # Updated function for text generation using the new API structure
 def generate_creative_text(prompt):
-    try:
-        response = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo",  # Change this to the model you prefer, e.g., "gpt-4" if available
-            messages=[
-                {"role": "system", "content": "You are a creative assistant."},
-                {"role": "user", "content": prompt}
-            ],
-            max_tokens=1024,
-            temperature=0.7,
-        )
-        return response['choices'][0]['message']['content'].strip()
-    except Exception as e:
-        return f"An error occurred during text generation: {str(e)}"
+    chat_completion = client.chat.completions.create(
+                messages=[
+                    {"role": "user", "content":prompt}
+                ],
+                model="llama-3.2-90b-text-preview"
+            )
+    chatbot_response = chat_completion.choices[0].message.content
+    return chatbot_response
 
 
 def process_audio(audio_path, image_option, creative_text_option):