Update app.py

app.py

@@ -1,31 +1,21 @@
 import os
-import openai
 import streamlit as st
 import subprocess
 from gtts import gTTS
 import cv2
+from huggingface_hub import InferenceClient
 
-openai.api_key = os.environ.get("openai_api_key")
+client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
 def generate_output(name, birth_date):
     if not birth_date:
         return None, "El campo de fecha de nacimiento es obligatorio."
 
     prompt = f"Tú horóscopo de hoy, si naciste el {birth_date}, es:"
-    response = openai.Completion.create(
-        engine="text-davinci-003",
-        prompt=prompt,
-        max_tokens=180,
-        temperature=0.6,
-        n=1,
-        stop=None,
-    )
-    gpt3_output = response.choices[0].text.strip()
+    response = client.text_generation(prompt, max_new_tokens=180, temperature=0.6)
+    gpt3_output = response[0]['generated_text'].strip()
     personalized_response = f"Tu horóscopo {name} nacido el {birth_date} es: {gpt3_output}"
     
-    if len(response.choices) == 0 or 'text' not in response.choices[0]:
-        return None, "No se pudo generar el texto."
-
     try:
         tts = gTTS(personalized_response, lang='es')
         audio_path = "audio.mp3"
@@ -54,5 +44,5 @@ if st.button("Generar Video"):
     if error_message:
         st.error(f"Error: {error_message}")
     else:
-            video_file = open(video_path, "rb").read()
-            st.video(video_file)
+        video_file = open(video_path, "rb").read()
+        st.video(video_file)