Update app.py

app.py

@@ -10,7 +10,7 @@ import re
 
 @st.cache_resource
 def get_inference_client():
-    return InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
+    return InferenceClient(model="mistralai/Mixtral-8x7B-Instruct-v0.1")
 
 def extract_and_clean_json(text):
     json_match = re.search(r'\{.*?\}', text, re.DOTALL)
@@ -45,7 +45,7 @@ def generate_presentation_content(topic, client):
     response = client.text_generation(prompt, max_new_tokens=2500, temperature=0.7)
     
     try:
-        json_str = extract_and_clean_json(response)
+        json_str = extract_and_clean_json(response['generated_text'])
         if json_str:
             slides_data = json.loads(json_str)
             if 'slides' in slides_data and len(slides_data['slides']) == 9:
@@ -55,130 +55,3 @@ def generate_presentation_content(topic, client):
         else:
             raise ValueError("No se encontró un JSON válido en la respuesta")
     except json.JSONDecodeError as e:
-        st.error(f"Error al decodificar JSON: {str(e)}")
-        st.text("JSON procesado (después de limpieza):")
-        st.code(json_str)
-        st.text("Respuesta original del modelo:")
-        st.code(response)
-        return None
-    except ValueError as e:
-        st.error(str(e))
-        st.text("Respuesta del modelo:")
-        st.code(response)
-        return None
-
-def apply_design(prs, design):
-    # ... [El código de la función apply_design permanece igual] ...
-def apply_design(prs, design):
-    if design == "Moderno":
-        background = prs.slides[0].background
-        fill = background.fill
-        fill.solid()
-        fill.fore_color.rgb = RGBColor(240, 240, 240)
-        
-        for slide in prs.slides:
-            for shape in slide.shapes:
-                if shape.has_text_frame:
-                    tf = shape.text_frame
-                    tf.text = tf.text
-                    for paragraph in tf.paragraphs:
-                        paragraph.font.color.rgb = RGBColor(0, 0, 0)
-                        if shape.name == 'Title':
-                            paragraph.font.size = Pt(44)
-                        else:
-                            paragraph.font.size = Pt(24)
-    elif design == "Corporativo":
-        background = prs.slides[0].background
-        fill = background.fill
-        fill.solid()
-        fill.fore_color.rgb = RGBColor(255, 255, 255)
-        
-        for slide in prs.slides:
-            left = top = Inches(0.5)
-            width = Inches(1)
-            height = Inches(0.5)
-            shape = slide.shapes.add_shape(MSO_SHAPE.RECTANGLE, left, top, width, height)
-            shape.fill.solid()
-            shape.fill.fore_color.rgb = RGBColor(0, 112, 192)
-            
-            for shape in slide.shapes:
-                if shape.has_text_frame:
-                    tf = shape.text_frame
-                    tf.text = tf.text
-                    for paragraph in tf.paragraphs:
-                        paragraph.font.color.rgb = RGBColor(0, 0, 0)
-                        if shape.name == 'Title':
-                            paragraph.font.size = Pt(40)
-                            paragraph.font.color.rgb = RGBColor(0, 112, 192)
-                        else:
-                            paragraph.font.size = Pt(20)
-    
-
-def create_powerpoint(slides, design):
-    prs = Presentation()
-    
-    # Diapositiva de título
-    slide = prs.slides.add_slide(prs.slide_layouts[0])
-    title = slide.shapes.title
-    subtitle = slide.placeholders[1]
-    title.text = slides[0]['title']
-    subtitle.text = slides[0]['content']
-
-    # Otras diapositivas
-    for slide_data in slides[1:]:
-        slide = prs.slides.add_slide(prs.slide_layouts[1])
-        title = slide.shapes.title
-        content = slide.placeholders[1]
-        title.text = slide_data['title']
-        content.text = slide_data['content']
-    
-    # Diapositiva "Gracias"
-    slide = prs.slides.add_slide(prs.slide_layouts[0])
-    title = slide.shapes.title
-    title.text = "¡Gracias!"
-    
-    apply_design(prs, design)
-    
-    pptx_buffer = io.BytesIO()
-    prs.save(pptx_buffer)
-    pptx_buffer.seek(0)
-    
-    return pptx_buffer
-
-def main():
-    st.title("Generador de presentaciones PowerPoint con IA")
-    
-    client = get_inference_client()
-    
-    topic = st.text_input("Por favor, ingrese el tema de la presentación:")
-    
-    design = st.selectbox(
-        "Seleccione un diseño para la presentación:",
-        ("Simple", "Moderno", "Corporativo")
-    )
-    
-    if st.button("Generar Presentación"):
-        if topic:
-            try:
-                with st.spinner("Generando contenido de la presentación..."):
-                    slides = generate_presentation_content(topic, client)
-                
-                if slides:
-                    with st.spinner("Creando archivo PowerPoint..."):
-                        pptx_buffer = create_powerpoint(slides, design)
-                    
-                    st.success("Presentación generada con éxito!")
-                    
-                    st.download_button(
-                        label="Descargar Presentación",
-                        data=pptx_buffer,
-                        file_name=f"{topic.replace(' ', '_')}_presentacion.pptx",
-                        mime="application/vnd.openxmlformats-officedocument.presentationml.presentation"
-                    )
-            except Exception as e:
-                st.error(f"Ocurrió un error al generar la presentación: {str(e)}")
-        else:
-            st.warning("Por favor, ingrese un tema para la presentación.")
-
-if __name__ == "__main__":
-    main()