Update app.py

app.py

@@ -59,9 +59,8 @@ Asegúrese de que la traducción sea precisa y conserve el significado original
     '''
     
     formatted_prompt = template.replace("{TEXT}", text).replace("{LANGUAGE}", target_language)
-    inputs = tokenizer(formatted_prompt, return_tensors="pt")
-    outputs = llm_engine_hf.invoke(formatted_prompt)
-    translated_text = outputs.content
+    response = llm_engine_hf.invoke(formatted_prompt)
+    translated_text = response['generated_text']
     
     return translated_text
 
@@ -74,9 +73,8 @@ def summarize(text, length):
 Asegúrese de que el resumen sea conciso y conserve el significado original del documento.
     '''
     
-    inputs = tokenizer(template, return_tensors="pt")
-    outputs = llm_engine_hf.invoke(template)
-    summarized_text = outputs.content
+    response = llm_engine_hf.invoke(template)
+    summarized_text = response['generated_text']
     
     return summarized_text
 
@@ -106,53 +104,60 @@ def handle_uploaded_file(uploaded_file):
     except Exception as e:
         return str(e)
 
-st.title("LexAIcon")
-st.write("Puedes conversar con este chatbot basado en Mistral7B-Instruct y subir archivos para que el chatbot los procese.")
-
-if "generated" not in st.session_state:
-    st.session_state["generated"] = []
-if "past" not in st.session_state:
-    st.session_state["past"] = []
-
-# Entrada del usuario
-user_input = st.text_input("Tú: ", "")
-
-# Opciones para la traducción
-target_language = st.selectbox("Selecciona el idioma de traducción", ["español", "inglés", "francés", "alemán"])
-
-# Opciones para el resumen
-summary_length = st.selectbox("Selecciona la longitud del resumen", ["corto", "medio", "largo"])
-
-# Manejo de archivos subidos
-uploaded_files = st.file_uploader("Sube un archivo", type=["txt", "pdf", "docx", "csv", "json"], accept_multiple_files=True)
-
-if st.button("Enviar"):
-    if user_input:
-        response = generate_response(user_input)
-        st.session_state.generated.append({"user": user_input, "bot": response})
-
-# Botones de Resumir, Traducir y Explicar
-operation = st.radio("Selecciona una operación", ["Resumir", "Traducir", "Explicar"])
-
-if st.button("Ejecutar"):
-    if uploaded_files:
-        for uploaded_file in uploaded_files:
-            file_content = handle_uploaded_file(uploaded_file)
-            if operation == "Resumir":
-                if summary_length == "corto":
-                    length = "de aproximadamente 50 palabras"
-                elif summary_length == "medio":
-                    length = "de aproximadamente 100 palabras"
-                elif summary_length == "largo":
-                    length = "de aproximadamente 500 palabras"
-                result = summarize(file_content, length)
-            elif operation == "Traducir":
-                result = translate(file_content, target_language)
-            elif operation == "Explicar":
-                result = classify_text(file_content)
-            st.write(result)
-
-if st.session_state.get("generated"):
-    for chat in st.session_state["generated"]:
-        st.write(f"Tú: {chat['user']}")
-        st.write(f"Chatbot: {chat['bot']}")
+def main():
+    st.title("LexAIcon")
+    st.write("Puedes conversar con este chatbot basado en Mistral7B-Instruct y subir archivos para que el chatbot los procese.")
+
+    if "generated" not in st.session_state:
+        st.session_state["generated"] = []
+    if "past" not in st.session_state:
+        st.session_state["past"] = []
+
+    # Entrada del usuario
+    user_input = st.text_input("Tú: ", "")
+
+    # Botones de Resumir, Traducir y Explicar
+    operation = st.radio("Selecciona una operación", ["Resumir", "Traducir", "Explicar"])
+
+    target_language = None
+    summary_length = None
+
+    if operation == "Traducir":
+        target_language = st.selectbox("Selecciona el idioma de traducción", ["español", "inglés", "francés", "alemán"])
+
+    if operation == "Resumir":
+        summary_length = st.selectbox("Selecciona la longitud del resumen", ["corto", "medio", "largo"])
+
+    # Manejo de archivos subidos
+    uploaded_files = st.file_uploader("Sube un archivo", type=["txt", "pdf", "docx", "csv", "json"], accept_multiple_files=True)
+
+    if st.button("Enviar"):
+        if user_input:
+            response = llm_engine_hf.invoke(user_input)
+            st.session_state.generated.append({"user": user_input, "bot": response['generated_text']})
+
+    if st.button("Ejecutar"):
+        if uploaded_files:
+            for uploaded_file in uploaded_files:
+                file_content = handle_uploaded_file(uploaded_file)
+                if operation == "Resumir":
+                    if summary_length == "corto":
+                        length = "de aproximadamente 50 palabras"
+                    elif summary_length == "medio":
+                        length = "de aproximadamente 100 palabras"
+                    elif summary_length == "largo":
+                        length = "de aproximadamente 500 palabras"
+                    result = summarize(file_content, length)
+                elif operation == "Traducir":
+                    result = translate(file_content, target_language)
+                elif operation == "Explicar":
+                    result = classify_text(file_content)
+                st.write(result)
+
+    if st.session_state.get("generated"):
+        for chat in st.session_state["generated"]:
+            st.write(f"Tú: {chat['user']}")
+            st.write(f"Chatbot: {chat['bot']}")
+
+if __name__ == "__main__":
+    main()