Update app.py

app.py

@@ -124,17 +124,29 @@ if submitted:
         input_text = make_query(description, question)
         try:
             input_ids = tokenizer.encode(input_text, return_tensors="pt")
-            with st.spinner("Генерация запроса..."):
-                outputs = model.generate(
-                    input_ids,
-                    max_length=200,
-                    num_beams=5,
-                    early_stopping=True,
-                    pad_token_id=tokenizer.eos_token_id,
-                )
-                generated_sql = tokenizer.decode(outputs[0], skip_special_tokens=True)
-                st.subheader("Результат:")
-                st.code(generated_sql, language="sql")
+            
+            animation_placeholder = st.empty()
+            
+            for frame in ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"]:
+                animation_placeholder.markdown(f"`{frame}` Подготовка к генерации...")
+                time.sleep(0.1)
+            
+            animation_placeholder.markdown("`⏳` Генерация SQL-запроса...")
+            outputs = model.generate(
+                input_ids,
+                max_length=200,
+                num_beams=5,
+                top_p=0.95,
+                early_stopping=True,
+                pad_token_id=tokenizer.eos_token_id,
+            )
+            
+            animation_placeholder.empty()
+
+            generated_sql = tokenizer.decode(outputs[0], skip_special_tokens=True)
+            st.subheader("Результат:")
+            st.code(generated_sql, language="sql")
+            
         except Exception as e:
             st.error(f"Ошибка при генерации: {str(e)}")
     else: