Upload app/services/file_generator.py with huggingface_hub

app/services/file_generator.py

@@ -1,64 +1,19 @@
-from typing import List
-from datetime import datetime
-from app.models.checklist import ChecklistItem
+from datetime import date
 
 
-class FileGenerator:
-    @staticmethod
-    def generate_markdown(
-        session_id: str,
-        checklist: List[ChecklistItem],
-        round_summaries: List[str]
-    ) -> str:
-        """Генерирует Markdown файл с результатами чеклиста"""
-        
-        date = datetime.now().strftime("%Y-%m-%d %H:%M")
-        
-        md_content = f"""# Чеклист созвона с клиентом
+def post_process_markdown(markdown: str, session_id: str) -> str:
+    """Clean up and ensure consistent markdown formatting."""
+    lines = markdown.strip().split("\n")
 
-**Дата:** {date}
-**Сессия:** {session_id}
+    # Ensure it starts with the expected header
+    if not lines[0].startswith("# "):
+        lines.insert(0, "# Чеклист созвона с клиентом")
 
----
+    content = "\n".join(lines)
 
-"""
-        # Группируем по категориям
-        categories = {}
-        for item in checklist:
-            if item.category not in categories:
-                categories[item.category] = []
-            categories[item.category].append(item)
+    # Ensure footer exists
+    footer = "\n---\n\n*Сгенерировано автоматически с помощью AI Checklist Agent*\n"
+    if "Сгенерировано автоматически" not in content:
+        content += footer
 
-        # Маппинг статусов на чекбоксы
-        status_map = {
-            "confirmed": "[x]",
-            "needs_clarification": "[ ] ⚠️",
-            "not_discussed": "[ ]"
-        }
-
-        for category, items in categories.items():
-            md_content += f"## {category}\n\n"
-            for item in items:
-                checkbox = status_map.get(item.status, "[ ]")
-                line = f"- {checkbox} {item.item}"
-                if item.notes:
-                    line += f" *({item.notes})*"
-                md_content += line + "\n"
-            md_content += "\n"
-
-        # Добавляем саммари раундов
-        if round_summaries:
-            md_content += "---\n\n## Саммари интервью\n\n"
-            for i, summary in enumerate(round_summaries, 1):
-                md_content += f"**Раунд {i}:** {summary}\n\n"
-
-        md_content += """---
-
-*Сгенерировано автоматически с помощью AI Checklist Agent*
-"""
-        
-        return md_content
-
-
-def get_file_generator() -> FileGenerator:
-    return FileGenerator()
+    return content