Add technical report generation feature

app.py

@@ -4,16 +4,50 @@ from PIL import Image
 import numpy as np
 import os
 from pathlib import Path
+from datetime import datetime
+import tempfile
 
 from model import RadarDetectionModel
 from feature_extraction import (calculate_amplitude, classify_amplitude,
                               calculate_distribution_range, classify_distribution_range,
                               calculate_attenuation_rate, classify_attenuation_rate,
-                              count_reflections, classify_reflections)
+                              count_reflections, classify_reflections,
+                              extract_features)
 from report_generation import generate_report, render_report
 from utils import plot_detection
 from database import save_report, get_report_history
 
+class TechnicalReportGenerator:
+    def __init__(self):
+        self.timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    
+    def _generate_technical_section(self, detection_result, features):
+        """Generate technical analysis section of the report."""
+        tech_doc = "## Technical Analysis\n\n"
+        
+        # Detection Results
+        tech_doc += "### Detection Results\n\n```python\n"
+        tech_doc += f"Confidence Scores: {detection_result['scores'].tolist()}\n"
+        tech_doc += f"Bounding Boxes: {detection_result['boxes'].tolist()}\n"
+        tech_doc += f"Labels: {detection_result['labels'].tolist()}\n"
+        tech_doc += "```\n\n"
+        
+        # Feature Analysis
+        tech_doc += "### Feature Analysis\n\n"
+        for feature_name, value in features.items():
+            tech_doc += f"- **{feature_name}**: {value}\n"
+        
+        # Signal Processing Details
+        tech_doc += "\n### Signal Processing Metrics\n\n"
+        tech_doc += "| Metric | Value | Classification |\n"
+        tech_doc += "|--------|--------|----------------|\n"
+        tech_doc += f"|Amplitude|{features['Amplitude']}|{classify_amplitude(features['Amplitude'])}|\n"
+        tech_doc += f"|Distribution Range|{features['Distribution Range']}|{features['Distribution Range']}|\n"
+        tech_doc += f"|Attenuation Rate|{features['Attenuation Rate']}|{features['Attenuation Rate']}|\n"
+        tech_doc += f"|Reflection Count|{features['Reflection Count']}|{features['Reflection Count']}|\n"
+        
+        return tech_doc
+
 # Initialize model with HF token from environment
 model = None
 try:
@@ -31,15 +65,15 @@ def initialize_model():
             return None, f"Error initializing model: {str(e)}"
     return model, None
 
-def process_image(image):
+def process_image(image, generate_tech_report=False):
     if image is None:
-        return None, "Please upload an image."
+        return None, "Please upload an image.", None
     
     # Initialize model if needed
     global model
     model, error = initialize_model()
     if error:
-        return None, error
+        return None, error, None
     
     try:
         # Convert to PIL Image if needed
@@ -78,18 +112,29 @@ def process_image(image):
         report = generate_report(detection_result, image, features)
         detection_image = plot_detection(image, detection_result)
         
+        # Generate technical report if requested
+        tech_report = None
+        if generate_tech_report:
+            report_gen = TechnicalReportGenerator()
+            tech_report = report_gen._generate_technical_section(detection_result, features)
+            
+            # Save technical report to a temporary file
+            with tempfile.NamedTemporaryFile(mode='w', delete=False, suffix='.md') as f:
+                f.write(tech_report)
+                tech_report = f.name
+        
         # Save report if database is configured
         try:
             save_report(report)
         except Exception as e:
             print(f"Warning: Could not save report: {str(e)}")
         
-        return detection_image, render_report(report)
+        return detection_image, render_report(report), tech_report
         
     except Exception as e:
         error_msg = f"Error processing image: {str(e)}"
         print(error_msg)
-        return None, error_msg
+        return None, error_msg, None
 
 def display_history():
     try:
@@ -127,11 +172,13 @@ with gr.Blocks(css=css) as iface:
     with gr.Row():
         with gr.Column(scale=1):
             input_image = gr.Image(type="pil", label="Upload Radar Image")
+            tech_report_checkbox = gr.Checkbox(label="Generate Technical Report", value=False)
             analyze_button = gr.Button("Analyze", variant="primary")
         
         with gr.Column(scale=2):
             output_image = gr.Image(type="pil", label="Detection Result")
             output_report = gr.HTML(label="Analysis Report")
+            tech_report_output = gr.File(label="Technical Report")
     
     with gr.Row():
         history_button = gr.Button("View History")
@@ -140,8 +187,8 @@ with gr.Blocks(css=css) as iface:
     # Set up event handlers
     analyze_button.click(
         fn=process_image,
-        inputs=[input_image],
-        outputs=[output_image, output_report]
+        inputs=[input_image, tech_report_checkbox],
+        outputs=[output_image, output_report, tech_report_output]
     )
     
     history_button.click(