Update app.py

app.py

@@ -1,64 +1,116 @@
 import gradio as gr
+import hashlib
 import os
-from pipeline import run_course_builder 
-
 from dotenv import load_dotenv
+
 load_dotenv()
 
+from pipeline import run_course_builder
+
+cache = {}
+
+def get_cache_key(query: str, toc: str, archive_name: str = "") -> str:
+    key_string = f"{query.strip()}|{toc.strip()}|{archive_name}"
+    return hashlib.md5(key_string.encode('utf-8')).hexdigest()
+
+
 def generate_course(query: str, toc: str, archive_file):
     if archive_file is None:
-        return "Пожалуйста, загрузите архив с материалами (.zip)", ""
-    
+        return "❌ Пожалуйста, загрузите ZIP-архив с материалами", ""
+
     try:
+        if hasattr(archive_file, 'read'):
+            file_content = archive_file.read()
+        else:
+            with open(str(archive_file), "rb") as f:
+                file_content = f.read()
+
+        archive_name = getattr(archive_file, 'name', 'unknown.zip')
+        cache_key = get_cache_key(query, toc, archive_name)
+
+        if cache_key in cache:
+            return "✅ Курс успешно сгенерирован!", cache[cache_key]
+
         temp_path = "temp_materials.zip"
         with open(temp_path, "wb") as f:
-            f.write(archive_file.read())
-        
-        result = run_course_builder(query, toc, temp_path)
-        
-        if isinstance(result, str):
-            markdown = result
-        else:
-            markdown = result.get("markdown", result.get("result", str(result)))
-        
+            f.write(file_content)
+
+        result = run_course_builder(query, toc or "", temp_path)
+
+        markdown = result if isinstance(result, str) else result.get("markdown") or result.get("result") or str(result)
+
+        cache[cache_key] = markdown
+
         return "✅ Курс успешно сгенерирован!", markdown
-        
+
     except Exception as e:
-        return f"❌ Ошибка: {str(e)}", ""
+        error_msg = f"❌ Ошибка: {str(e)}"
+        return error_msg, error_msg
+
+
+def clear_cache():
+    cache.clear()
+    if os.path.exists("temp_materials.zip"):
+        try:
+            os.remove("temp_materials.zip")
+        except:
+            pass
+    return "🗑️ Кэш очищен", ""
 
-# ====================== Gradio Interface ======================
 
+def clear_cache():
+    cache.clear()
+    if os.path.exists("temp_materials.zip"):
+        try:
+            os.remove("temp_materials.zip")
+        except:
+            pass
+    return "🗑️ Кэш и временные файлы очищены", ""
+
+
+# ====================== Gradio Interface ======================
 with gr.Blocks(title="Генератор учебных курсов", theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 📚 Генератор Интенсивных Учебных Курсов")
-    gr.Markdown("Загрузи оглавление + архив с материалами → получи красивый понедельный план")
+    gr.Markdown("Загрузите оглавление и архив с материалами — получите готовый понедельный план")
 
     with gr.Row():
         with gr.Column(scale=2):
             query_input = gr.Textbox(
-                label="Запрос / Описание курса",
-                placeholder="Напиши мне интенсивный курс на 9 недель по алгебре и введению в алгебраическую геометрию...",
+                label="Описание курса",
+                placeholder="Напишите, какой курс вам нужен (например: интенсивный курс на 9 недель по алгебре и введению в алгебраическую геометрию)...",
                 lines=3
             )
             toc_input = gr.Textbox(
-                label="Оглавление (TOC)",
-                placeholder="Вставь оглавление курса...",
-                lines=10
+                label="Оглавление (необязательно)",
+                placeholder="Вставьте оглавление курса...",
+                lines=8
             )
             file_input = gr.File(
                 label="Архив с материалами (.zip)",
-                file_types=[".zip"]
+                file_types=[".zip"],
+                type="filepath"
             )
 
         with gr.Column(scale=3):
-            output_status = gr.Textbox(label="Статус", interactive=False)
-            output_markdown = gr.Markdown(label="Сгенерированный курс", height=800)
+            status_output = gr.Textbox(label="Статус", interactive=False)
+            markdown_output = gr.Markdown(label="Сгенерированный курс", height=720)
 
-    btn = gr.Button("🚀 Сгенерировать курс", variant="primary", size="large")
+    with gr.Row():
+        generate_btn = gr.Button("🚀 Сгенерировать курс", variant="primary", size="large")
+        clear_btn = gr.Button("🗑️ Очистить кэш и файлы", variant="secondary")
 
-    btn.click(
+    # Обработка генерации
+    generate_btn.click(
         fn=generate_course,
         inputs=[query_input, toc_input, file_input],
-        outputs=[output_status, output_markdown]
+        outputs=[status_output, markdown_output]
+    )
+
+    # Очистка кэша
+    clear_btn.click(
+        fn=clear_cache,
+        inputs=[],
+        outputs=[status_output, markdown_output]
     )
 
 if __name__ == "__main__":