AlejandraFlorezM commited on
Commit
3b73b92
·
verified ·
1 Parent(s): c593953

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -104
app.py DELETED
@@ -1,104 +0,0 @@
1
- import streamlit as st
2
- import random
3
- from transformers import pipeline
4
-
5
- # Configuración de la página
6
- st.set_page_config(
7
- page_title="¿No sabes qué decir? Te ayudamos",
8
- page_icon="💡",
9
- layout="centered"
10
- )
11
-
12
- # Estilo de la app
13
- st.markdown(
14
- """
15
- <style>
16
- body {
17
- background-color: #fefaf4;
18
- }
19
- .title {
20
- font-size: 38px;
21
- font-weight: bold;
22
- text-align: center;
23
- color: #a3b18a;
24
- margin-bottom: 10px;
25
- }
26
- .subtitle {
27
- font-size: 18px;
28
- text-align: center;
29
- color: #588157;
30
- }
31
- .footer {
32
- text-align: center;
33
- font-size: 12px;
34
- margin-top: 30px;
35
- color: #8d99ae;
36
- }
37
- .stButton>button {
38
- background-color: #ffdab9;
39
- color: #3d405b;
40
- font-size: 16px;
41
- }
42
- </style>
43
- """,
44
- unsafe_allow_html=True
45
- )
46
-
47
- # Diccionario de conectores
48
- conectores = {
49
- "Explicación": ["porque", "ya que", "puesto que", "dado que"],
50
- "Contraste": ["sin embargo", "aunque", "no obstante", "a pesar de"],
51
- "Adición": ["además", "asimismo", "igualmente", "por otra parte"],
52
- "Causalidad negativa": ["pero", "sin embargo", "aunque"]
53
- }
54
-
55
- # Seleccionar conector lógico aleatorio
56
- def agregar_conector_automatico(texto_usuario):
57
- categoria = random.choice(list(conectores.keys()))
58
- conector = random.choice(conectores[categoria])
59
- # Si el texto no termina con un conector lógico, lo añadimos
60
- if not texto_usuario.strip().endswith(tuple(conector for sublist in conectores.values() for conector in sublist)):
61
- texto_usuario += f" {conector}"
62
- return texto_usuario
63
-
64
- # Cargar modelo
65
- @st.cache_resource
66
- def load_model():
67
- return pipeline("text-generation", model="datificate/gpt2-small-spanish")
68
-
69
- generator = load_model()
70
-
71
- # Título llamativo
72
- st.markdown('<p class="title">¿No sabes qué decir? 💡</p>', unsafe_allow_html=True)
73
- st.markdown('<p class="subtitle">¡Te ayudamos a completar tus ideas con inteligencia artificial!</p>', unsafe_allow_html=True)
74
-
75
- # Entrada del usuario
76
- st.subheader("📝 Escribe tu idea:")
77
- prompt = st.text_area("Completa tu idea:", placeholder="Ejemplo: Me gustaría escribir una carta a...")
78
-
79
- # Generar respuesta
80
- if st.button("✨ Generar Texto ✨"):
81
- if not prompt.strip():
82
- st.warning("⚠️ Por favor, ingresa un texto válido.")
83
- else:
84
- with st.spinner("⏳ Generando texto..."):
85
- # Añadir un conector lógico automáticamente si falta
86
- prompt_con_conector = agregar_conector_automatico(prompt)
87
- result = generator(
88
- prompt_con_conector,
89
- max_new_tokens=30,
90
- temperature=0.2,
91
- top_p=0.6,
92
- top_k=10,
93
- repetition_penalty=1.5,
94
- num_return_sequences=1,
95
- eos_token_id=50256,
96
- do_sample=True
97
- )
98
- raw_response = result[0]["generated_text"]
99
- response = truncar_en_punto(raw_response) #cortar en el primer punto final
100
- st.subheader("🖋️ Texto Generado:")
101
- st.success(response)
102
-
103
- # Footer
104
- st.markdown('<p class="footer">Grupo 7 - Procesamiento de datos con AI - Especialización de Inteligencia Artificial UAO</p>', unsafe_allow_html=True)