Update report_generator.py

report_generator.py

@@ -5,7 +5,6 @@ from pathlib import Path
 import logging
 from datetime import datetime
 import matplotlib.pyplot as plt
-import seaborn as sns
 from docx import Document
 from docx.shared import Inches, Pt
 from docx.enum.text import WD_ALIGN_PARAGRAPH
@@ -21,11 +20,11 @@ class ReportGenerator:
         self.output_dir.mkdir(exist_ok=True)
         self.logger = logging.getLogger(__name__)
 
-    def generate_excel_report(
+    async def generate_excel_report(
         self,
         analyses: Dict[str, 'OfferAnalysis'],
         criteria: List['EvaluationCriterion']
-    ) -> str:
+    ) -> pd.DataFrame:
         """
         Generuje raport Excel z wynikami oceny.
         
@@ -34,7 +33,7 @@ class ReportGenerator:
             criteria: Lista kryteriów oceny
             
         Returns:
-            str: Ścieżka do wygenerowanego pliku Excel
+            pd.DataFrame: DataFrame z wynikami
         """
         try:
             # Przygotuj dane do DataFrame
@@ -93,13 +92,13 @@ class ReportGenerator:
                     worksheet.set_column(i+1, i+1, 50)  # Uzasadnienie
             
             self.logger.info(f"Wygenerowano raport Excel: {excel_path}")
-            return str(excel_path)
+            return df
 
         except Exception as e:
             self.logger.error(f"Błąd podczas generowania raportu Excel: {str(e)}")
             raise
 
-    def generate_word_report(
+    async def generate_word_report(
         self,
         analyses: Dict[str, 'OfferAnalysis'],
         criteria: List['EvaluationCriterion'],
@@ -114,34 +113,23 @@ class ReportGenerator:
             comparison_results: Wyniki porównania ofert
             
         Returns:
-            str: Ścieżka do wygenerowanego pliku Word
+            str: Markdown z raportem
         """
         try:
-            doc = Document()
-            
-            # 1. Strona tytułowa
-            title = doc.add_heading('Raport z Oceny Ofert Przetargowych', 0)
-            title.alignment = WD_ALIGN_PARAGRAPH.CENTER
-            
-            doc.add_paragraph(f'Data utworzenia: {datetime.now().strftime("%d.%m.%Y")}')
-            doc.add_paragraph(f'Liczba ocenionych ofert: {len(analyses)}')
+            # Generuj raport w formacie Markdown
+            report_lines = []
             
-            # 2. Spis treści
-            doc.add_page_break()
-            doc.add_heading('Spis treści', 1)
-            # TODO: Dodać automatyczny spis treści
+            # 1. Nagłówek
+            report_lines.append("# Raport z Oceny Ofert Przetargowych")
+            report_lines.append(f"Data utworzenia: {datetime.now().strftime('%d.%m.%Y')}")
+            report_lines.append(f"Liczba ocenionych ofert: {len(analyses)}")
             
-            # 3. Podsumowanie wykonawcze
-            doc.add_page_break()
-            doc.add_heading('Podsumowanie wykonawcze', 1)
+            # 2. Podsumowanie wykonawcze
+            report_lines.append("\n## Podsumowanie wykonawcze")
             
             # Tabela z kluczowymi wynikami
-            summary_table = doc.add_table(rows=1, cols=3)
-            summary_table.style = 'Table Grid'
-            header_cells = summary_table.rows[0].cells
-            header_cells[0].text = 'Oferent'
-            header_cells[1].text = 'Wynik końcowy'
-            header_cells[2].text = 'Ranking'
+            report_lines.append("\n| Oferent | Wynik końcowy | Ranking |")
+            report_lines.append("|----------|---------------|---------|")
             
             # Posortuj oferty wg wyniku
             sorted_offers = sorted(
@@ -151,103 +139,80 @@ class ReportGenerator:
             )
             
             for i, (offer_id, analysis) in enumerate(sorted_offers, 1):
-                row_cells = summary_table.add_row().cells
-                row_cells[0].text = offer_id
-                row_cells[1].text = f"{analysis.total_score:.2f}"
-                row_cells[2].text = str(i)
+                report_lines.append(
+                    f"| {offer_id} | {analysis.total_score:.2f} | {i} |"
+                )
             
-            # 4. Szczegółowa analiza każdej oferty
+            # 3. Szczegółowa analiza ofert
             for offer_id, analysis in analyses.items():
-                doc.add_page_break()
-                doc.add_heading(f'Analiza oferty: {offer_id}', 1)
+                report_lines.append(f"\n## Analiza oferty: {offer_id}")
+                report_lines.append(f"Wynik końcowy: {analysis.total_score:.2f}/100 punktów")
                 
-                # Wynik końcowy
-                doc.add_paragraph(
-                    f'Wynik końcowy: {analysis.total_score:.2f}/100 punktów'
-                )
-                
-                # Mocne strony
-                doc.add_heading('Mocne strony:', 2)
+                report_lines.append("\n### Mocne strony:")
                 for strength in analysis.strengths:
-                    doc.add_paragraph(strength, style='List Bullet')
+                    report_lines.append(f"* {strength}")
                 
-                # Słabe strony
-                doc.add_heading('Słabe strony:', 2)
+                report_lines.append("\n### Słabe strony:")
                 for weakness in analysis.weaknesses:
-                    doc.add_paragraph(weakness, style='List Bullet')
-                
-                # Szczegółowe oceny
-                doc.add_heading('Szczegółowe oceny:', 2)
-                scores_table = doc.add_table(rows=1, cols=3)
-                scores_table.style = 'Table Grid'
+                    report_lines.append(f"* {weakness}")
                 
-                header_cells = scores_table.rows[0].cells
-                header_cells[0].text = 'Kryterium'
-                header_cells[1].text = 'Ocena'
-                header_cells[2].text = 'Uzasadnienie'
+                report_lines.append("\n### Szczegółowe oceny:")
+                report_lines.append("\n| Kryterium | Ocena | Uzasadnienie |")
+                report_lines.append("|-----------|--------|--------------|")
                 
                 for eval in analysis.evaluations:
-                    row_cells = scores_table.add_row().cells
-                    row_cells[0].text = eval.criterion_name
-                    row_cells[1].text = f"{eval.score:.2f}"
-                    row_cells[2].text = eval.justification
+                    report_lines.append(
+                        f"| {eval.criterion_name} | {eval.score:.2f} | {eval.justification} |"
+                    )
             
-            # 5. Porównanie ofert
-            doc.add_page_break()
-            doc.add_heading('Porównanie ofert', 1)
-            doc.add_paragraph(comparison_results)
-            
-            # 6. Zapisz dokument
-            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-            word_path = self.output_dir / f"raport_oceny_{timestamp}.docx"
-            doc.save(word_path)
+            # 4. Porównanie ofert
+            report_lines.append("\n## Porównanie ofert")
+            report_lines.append(str(comparison_results))
             
-            self.logger.info(f"Wygenerowano raport Word: {word_path}")
-            return str(word_path)
+            report = "\n".join(report_lines)
+            self.logger.info("Wygenerowano raport Word w formacie Markdown")
+            return report
 
         except Exception as e:
             self.logger.error(f"Błąd podczas generowania raportu Word: {str(e)}")
             raise
 
-    def generate_visualizations(
-        self,
-        analyses: Dict[str, 'OfferAnalysis']
-    ) -> List[str]:
+    def _generate_bar_plot(self, analyses: Dict[str, 'OfferAnalysis']) -> str:
         """
-        Generuje wizualizacje wyników dla raportów.
+        Generuje wykres słupkowy porównujący wyniki końcowe.
         
         Args:
             analyses: Słownik z analizami ofert
             
         Returns:
-            List[str]: Lista ścieżek do wygenerowanych wykresów
+            str: Ścieżka do zapisanego wykresu
         """
         try:
-            plot_paths = []
+            # Przygotuj dane
+            offer_ids = list(analyses.keys())
+            scores = [analysis.total_score for analysis in analyses.values()]
             
-            # 1. Wykres radarowy dla wszystkich ofert
-            self._generate_radar_plot(analyses)
+            # Utwórz wykres
+            plt.figure(figsize=(10, 6))
+            plt.bar(offer_ids, scores)
             
-            # 2. Wykres słupkowy porównawczy
-            self._generate_bar_plot(analyses)
+            # Dodaj etykiety
+            plt.title('Porównanie wyników końcowych')
+            plt.xlabel('Oferenci')
+            plt.ylabel('Wynik końcowy')
+            plt.xticks(rotation=45)
             
-            # 3. Heatmapa ocen
-            self._generate_heatmap(analyses)
+            # Dodaj wartości nad słupkami
+            for i, score in enumerate(scores):
+                plt.text(i, score, f'{score:.1f}', ha='center', va='bottom')
+            
+            # Zapisz wykres
+            plot_path = self.output_dir / 'wyniki_koncowe.png'
+            plt.savefig(plot_path, bbox_inches='tight')
+            plt.close()
+            
+            return str(plot_path)
             
-            return plot_paths
-
         except Exception as e:
-            self.logger.error(f"Błąd podczas generowania wizualizacji: {str(e)}")
-            return []
-
-    def _generate_radar_plot(self, analyses: Dict[str, 'OfferAnalysis']) -> str:
-        """Generuje wykres radarowy porównujący oferty"""
-        pass  # TODO: Implementacja wykresu radarowego
-
-    def _generate_bar_plot(self, analyses: Dict[str, 'OfferAnalysis']) -> str:
-        """Generuje wykres słupkowy porównujący wyniki końcowe"""
-        pass  # TODO: Implementacja wykresu słupkowego
-
-    def _generate_heatmap(self, analyses: Dict[str, 'OfferAnalysis']) -> str:
-        """Generuje heatmapę ocen dla wszystkich kryteriów"""
-        pass  # TODO: Implementacja heatmapy
+            self.logger.error(f"Błąd podczas generowania wykresu: {str(e)}")
+            return ""