Добавлены правила отслеживания Git LFS для файлов .xlsx

.gitattributes

@@ -35,3 +35,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 final_data_wb.xlsx filter=lfs diff=lfs merge=lfs -text
 data/final_data_ozon.xlsx filter=lfs diff=lfs merge=lfs -text
+*.xlsx filter=lfs diff=lfs merge=lfs -text

app.py

@@ -23,14 +23,16 @@ def main():
 
     db = initialize()
 
-    data_files = ['data/final_data_wb.xlsx', 'data/final_data_ozon.xlsx']
+    data_files = ['data/final_data_ozon_with_names.xlsx', 'data/final_data_wb_with_names.xlsx']
     data = load_data(data_files)
 
+    # Initialize session state variables
     if 'original_review' not in st.session_state:
         st.session_state.original_review = ''
     if 'user_name' not in st.session_state:
         st.session_state.user_name = ''
-
+    if 'product_name' not in st.session_state:
+        st.session_state.product_name = ''  # New variable for product name
     if 'model_a' not in st.session_state:
         st.session_state.model_a = ''
     if 'model_b' not in st.session_state:
@@ -41,6 +43,8 @@ def main():
         st.session_state.rewrite_b = ''
     if 'vote_submitted' not in st.session_state:
         st.session_state.vote_submitted = False
+    if 'used_models' not in st.session_state:
+        st.session_state.used_models = []
 
     instruction_gpt = read_prompt('prompts/gpt_prompt.txt')
     instruction_gemma = read_prompt('prompts/gemma_prompt.txt')
@@ -49,33 +53,40 @@ def main():
         random_row = get_random_review(data)
         st.session_state.original_review = random_row['body']
         st.session_state.user_name = random_row['user_name']
+        st.session_state.product_name = random_row['names']  # Get the product name
+
         if pd.isna(st.session_state.user_name) or st.session_state.user_name.strip() == '':
             st.session_state.user_name = 'Пользователь'
+        if pd.isna(st.session_state.product_name) or st.session_state.product_name.strip() == '':
+            st.session_state.product_name = 'товар'  # Default value if name is missing
+
+        st.session_state.used_models = []
 
         selected_models = random.sample(models, 2)
         st.session_state.model_a = selected_models[0]
         st.session_state.model_b = selected_models[1]
+        st.session_state.used_models.extend(selected_models)
+
         st.session_state.rewrite_a = ''
         st.session_state.rewrite_b = ''
         st.session_state.vote_submitted = False
 
     if st.session_state.original_review:
-        st.markdown(f"<h2>📝 Оригинальный отзыв от {st.session_state.user_name}:</h2>", unsafe_allow_html=True)
+        st.markdown(f"<h2>📝 Оригинальный отзыв от {st.session_state.user_name} о {st.session_state.product_name}:</h2>", unsafe_allow_html=True)
         st.markdown(
             f"<div class='original-review'>{st.session_state.original_review}</div>",
             unsafe_allow_html=True
         )
 
-
         if st.button("⚔️ Сгенерировать рерайты"):
             with st.spinner("Генерация рерайтов..."):
 
                 def generate_rewrite_for_model(model_name, text):
                     if "gemma" in model_name.lower():
-                        instruction = instruction_gemma.format(user_text=text)
+                        instruction = instruction_gemma.format(user_text=text, product_name=st.session_state.product_name)
                         return get_gemma_response(model_name, instruction)
                     else:  
-                        instruction = instruction_gpt.format(user_name=st.session_state.user_name)
+                        instruction = instruction_gpt.format(user_name=st.session_state.user_name, product_name=st.session_state.product_name)
                         return generate_rewrite(model_name, instruction, text)
 
                 st.session_state.rewrite_a = generate_rewrite_for_model(
@@ -89,7 +100,7 @@ def main():
             <p>Выбери рерайт, который соответствует данным критериям:</p>
             <ol>
                 <li>Сохранен смысл оригинального отзыва, но написан другими словами.</li>
-                <li>Рерайт соответсвует полу, как в оригинале.</li>
+                <li>Рерайт соответствует полу, как в оригинале.</li>
                 <li>Написан повседневным языком и не похож на сгенерированный.</li>
                 <li>Имеет структуру <strong>Отзыв</strong>, <strong>Преимущества</strong>, <strong>Недостатки</strong>.</li>
             </ol>
@@ -120,6 +131,37 @@ def main():
                     st.success("Вы выбрали Model B")
                     save_vote(db, 'right', models)
 
+        if not st.session_state.vote_submitted:
+            if st.button("😕 Ни один рерайт не понравился, получить другие рерайты"):
+                available_models = [model for model in models if model not in st.session_state.used_models]
+                if len(available_models) < 2:
+                    st.session_state.used_models = []
+                    available_models = models.copy()
+                selected_models = random.sample(available_models, 2)
+                st.session_state.model_a = selected_models[0]
+                st.session_state.model_b = selected_models[1]
+                st.session_state.used_models.extend(selected_models)
+
+                st.session_state.rewrite_a = ''
+                st.session_state.rewrite_b = ''
+                st.session_state.vote_submitted = False
+
+                with st.spinner("Генерация новых рерайтов..."):
+
+                    def generate_rewrite_for_model(model_name, text):
+                        if "gemma" in model_name.lower():
+                            instruction = instruction_gemma.format(user_text=text, product_name=st.session_state.product_name)
+                            return get_gemma_response(model_name, instruction)
+                        else:  
+                            instruction = instruction_gpt.format(user_name=st.session_state.user_name, product_name=st.session_state.product_name)
+                            return generate_rewrite(model_name, instruction, text)
+
+                    st.session_state.rewrite_a = generate_rewrite_for_model(
+                        st.session_state.model_a, st.session_state.original_review)
+                    st.session_state.rewrite_b = generate_rewrite_for_model(
+                        st.session_state.model_b, st.session_state.original_review)
+                st.experimental_rerun()
+
 if __name__ == '__main__':
     main()
 
@@ -131,3 +173,4 @@ if __name__ == '__main__':
 
 
 
+

data/{final_data_ozon.xlsx → final_data_ozon_with_names.xlsx}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ae5cb36370a55dc85acbdd2fdb21876e4b446939d041765a80a25f9e23d48834
-size 1848234
+oid sha256:ffcc729bbcce05c04402549efb0bc26b1881e69a10f8302cc3d1135dda376d8d
+size 2367315

data/{final_data_wb.xlsx → final_data_wb_with_names.xlsx}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b65f16124ffc66b8eacdd1654d3348ad599fffe837ef356045ea6bca6ae5a8d9
-size 1985886
+oid sha256:317af13e5c2d1769ad5e22b1dfd0f06b283c1a5bc6b0dc3081c2dc483f2f0c67
+size 2367315

prompts/gemma_prompt.txt

@@ -4,8 +4,9 @@
 Твоя задача — переписать предоставленный текст, сохранив его смысл и основные знаки препинания, которые встречаются в оригинальном тексте, 
 но используя другие слова. Используй повседневный стиль речи, чтобы текст выглядел естественно и не был похож на сгенерированный. 
 Избегай грамматических и стилистических ошибок, грубостей, матных слов и жаргона. Не искажай факты и названия товаров.
- 
 
+**Название товара:** {product_name}
+ 
 **Строго соблюдай следующий формат ответа и не добавляй ничего лишнего:**
 
 Отзыв: [текст отзыва]
@@ -24,3 +25,4 @@
 - Игнорируй всё остальное и не добавляй дополнительных комментариев.
 <end_of_turn>
 <start_of_turn>model
+

prompts/gpt_prompt.txt

@@ -1,13 +1,15 @@
-**Ты профессиональный рерайтер**. Твоя задача — переосмыслить предоставленный текст, и написать качественный рерайт, сохранив 
-смысл из оргинального отзыва, но переписав его другими словами. 
-Сохрани смысл текста, основные знаки препинания, которые встречаются в оригинале текста.
+**Ты профессиональный рерайтер**. Твоя задача — переосмыслить предоставленный текст и написать качественный рерайт, сохранив 
+смысл из оригинального отзыва, но переписав его другими словами. 
+Сохрани смысл текста и основные знаки препинания, которые встречаются в оригинале текста.
 Используй повседневный стиль речи и избегай стилистических и 
-грамматических ошибок. 
+грамматических ошибок.
 
 Твой рерайт должен быть **естественным и человечным**. 
-Особое внимание удели тому, чтобы не искажать факты, особенно названия товаров и конкретные детали. 
+Особое внимание удели тому, чтобы не искажать факты, особенно названия товаров и конкретные детали.
 
-**Ты должен добавить в текст немного повседневной речи и **cленга**, чтобы он выглядел натуральным и не был похож на сгенерированный.**
+**Название товара:** {product_name}
+
+**Ты должен добавить в текст немного повседневной речи и **сленга**, чтобы он выглядел натуральным и не был похож на сгенерированный.**
 **Важно:** Автор отзыва — **{user_name}**. Постарайся сохранить стиль и тон, соответствующий полу и возрасту автора.
 
 **Пример:**
@@ -17,10 +19,10 @@
 - Входной контент подается в формате строки:
     "Этот планшет просто отличный! Легкое перо, множество настраиваемых кнопок и большой, четкий экран."
 
-- На выходе ожидается текст, с выполнеными критериями:
+- На выходе ожидается текст, с выполненными критериями:
     - Отзыв переписан с сохранением смысла, но сформулирован другими словами.
     - Нет грамматических или логических ошибок.
-    - Добавлена повседневная речь и сленг
+    - Добавлена повседневная речь и сленг.
     - Отзыв должен быть переписан естественным языком, не похож на сгенерированный!
     - Рерайт должен быть структурирован так, чтобы каждый тег начинался с новой строки, и между ними была пустая строка. Используйте следующий формат:
     Отзыв: [текст отзыва]
@@ -38,21 +40,9 @@
 Недостатки: нет
 
 Рерайт: 
-Отзыв: Я просто обожаю обертывания, хотя иногда лень их делать. Но когда аромат шоколада наполняет воздух, то мотивация появляется сама собой! 
-🔥 Честно говоря, мне больше нравится горячее обертывание, оно как теплый плед, а не холодный душ. 🥶
+Отзыв: Я просто обожаю обёртывания, хотя иногда лень их делать. Но когда аромат шоколада наполняет воздух, то мотивация появляется сама собой! 
+🔥 Честно говоря, мне больше нравится горячее обёртывание, оно как тёплый плед, а не холодный душ. 🥶
 Преимущества: очень вкусно пахнет шоколадом
 Недостатки: нет
 
-Оригинал: 
-Отзыв: по началу руль казался толстым , как сарделька, но быстро привыкаешь.однозначно рекомендую - она крута!
-Преимущества: оплетка - бомба! очень красивая и удобная. руки не скользят , мягкая и прятная на ощупь.
-Недостатки: на mitsubishi outlander натянула с трудом, но, наверное, так и надо, чтоб не прокручивала и не болталась. заказала такую же подруге, 
-через день выпала страза, хотя, у меня всё норм - запасайтесь супер клеем.
-
-Рерайт:
-Отзыв: Сначала руль с этой оплеткой показался мне толстоватым, как батон колбасы, но привыкаешь очень быстро. В общем, советую всем – вещь классная!
-Преимущества: Сама оплетка просто супер! И красивая, и удобная. Руки не скользят, материал мягкий и приятный.
-Недостатки: На мой Mitsubishi Outlander натягивалась туговато, но, видимо, так и должно быть, чтобы не прокручивалась и не болталась. 
-Заказала такую же подруге, у нее через день отвалился стразик, хотя у меня все ок. Так что, на всякий случай, держите под рукой суперклей.
-
-**Следуй этим инструкциям внимательно, чтобы переписанный текст выглядел максимально естественно и качественно.**
+**Следуй этим инструкциям внимательно, чтобы переписанный текст выглядел максимально естественно и качественно.**