ALwrity AI Blog Writer - Added Google Grounding UI Implementation

frontend/src/services/blogWriterApi.ts

@@ -1,4 +1,4 @@
-import { apiClient, aiApiClient, longRunningApiClient } from "../api/client";
+import { apiClient, aiApiClient, longRunningApiClient, pollingApiClient } from "../api/client";
 
 export interface PersonaInfo {
   persona_id?: string;
@@ -13,6 +13,8 @@ export interface ResearchSource {
   excerpt?: string;
   credibility_score?: number;
   published_at?: string;
+  index?: number;
+  source_type?: string;
 }
 
 export interface BlogResearchRequest {
@@ -25,14 +27,47 @@ export interface BlogResearchRequest {
   persona?: PersonaInfo;
 }
 
+export interface GroundingChunk {
+  title: string;
+  url: string;
+  confidence_score?: number;
+}
+
+export interface GroundingSupport {
+  confidence_scores: number[];
+  grounding_chunk_indices: number[];
+  segment_text: string;
+  start_index?: number;
+  end_index?: number;
+}
+
+export interface Citation {
+  citation_type: string;
+  start_index: number;
+  end_index: number;
+  text: string;
+  source_indices: number[];
+  reference: string;
+}
+
+export interface GroundingMetadata {
+  grounding_chunks: GroundingChunk[];
+  grounding_supports: GroundingSupport[];
+  citations: Citation[];
+  search_entry_point?: string;
+  web_search_queries: string[];
+}
+
 export interface BlogResearchResponse {
   success: boolean;
+  keywords?: string[];
   sources: ResearchSource[];
   keyword_analysis: Record<string, any>;
   competitor_analysis: Record<string, any>;
   suggested_angles: string[];
   search_widget?: string;
   search_queries?: string[];
+  grounding_metadata?: GroundingMetadata;
   error_message?: string;
 }
 
@@ -46,10 +81,81 @@ export interface BlogOutlineSection {
   keywords: string[];
 }
 
+export interface SourceMappingStats {
+  total_sources_mapped: number;
+  coverage_percentage: number;
+  average_relevance_score: number;
+  high_confidence_mappings: number;
+}
+
+export interface GroundingInsights {
+  confidence_analysis?: {
+    average_confidence: number;
+    confidence_distribution: { high: number; medium: number; low: number };
+    high_confidence_sources_count: number;
+    high_confidence_insights: string[];
+  };
+  authority_analysis?: {
+    average_authority_score: number;
+    authority_distribution: { high: number; medium: number; low: number };
+    high_authority_sources: Array<{ title: string; url: string; score: number }>;
+  };
+  temporal_analysis?: {
+    recency_score: number;
+    trending_topics: string[];
+    temporal_balance: string;
+  };
+  content_relationships?: {
+    related_concepts: string[];
+    content_gaps: string[];
+    concept_coverage_score: number;
+    gap_count: number;
+  };
+  citation_insights?: {
+    total_citations: number;
+    citation_types: Record<string, number>;
+    citation_density: number;
+    citation_quality_score: number;
+  };
+  search_intent_insights?: {
+    primary_intent: string;
+    user_questions: string[];
+    intent_signals_count: number;
+  };
+  quality_indicators?: {
+    overall_quality_score: number;
+    quality_grade: string;
+    key_quality_factors: {
+      confidence: number;
+      authority: number;
+      citations: number;
+      coverage: number;
+    };
+  };
+}
+
+export interface OptimizationResults {
+  overall_quality_score: number;
+  improvements_made: string[];
+  optimization_focus: string;
+}
+
+export interface ResearchCoverage {
+  sources_utilized: number;
+  content_gaps_identified: number;
+  competitive_advantages: string[];
+}
+
 export interface BlogOutlineResponse {
   success: boolean;
   title_options: string[];
   outline: BlogOutlineSection[];
+  
+  // Additional metadata for enhanced UI
+  source_mapping_stats?: SourceMappingStats;
+  grounding_insights?: GroundingInsights;
+  optimization_results?: OptimizationResults;
+  research_coverage?: ResearchCoverage;
 }
 
 export interface BlogSectionResponse {
@@ -86,23 +192,48 @@ export interface BlogPublishResponse {
   post_id?: string;
 }
 
+export interface TaskStatusResponse {
+  task_id: string;
+  status: 'pending' | 'running' | 'completed' | 'failed';
+  created_at: string;
+  progress_messages: Array<{
+    timestamp: string;
+    message: string;
+  }>;
+  result?: BlogResearchResponse;
+  error?: string;
+}
+
 export const blogWriterApi = {
-  async research(payload: BlogResearchRequest): Promise<BlogResearchResponse> {
-    // Use the direct research endpoint for simplicity
-    const { data } = await apiClient.post("/api/blog/research", payload);
+  // Async polling endpoints
+  async startResearch(payload: BlogResearchRequest): Promise<{task_id: string; status: string}> {
+    const { data } = await apiClient.post("/api/blog/research/start", payload);
     return data;
   },
 
+  async pollResearchStatus(taskId: string): Promise<TaskStatusResponse> {
+    console.log('Polling research status for task:', taskId);
+    const { data } = await pollingApiClient.get(`/api/blog/research/status/${taskId}`);
+    console.log('Research status response:', data);
+    return data;
+  },
+
+  async startOutlineGeneration(payload: { research: BlogResearchResponse; persona?: PersonaInfo; word_count?: number; custom_instructions?: string }): Promise<{task_id: string; status: string}> {
+    const { data } = await aiApiClient.post("/api/blog/outline/start", payload);
+    return data;
+  },
+
+  async pollOutlineStatus(taskId: string): Promise<TaskStatusResponse> {
+    const { data } = await pollingApiClient.get(`/api/blog/outline/status/${taskId}`);
+    return data;
+  },
+
+
   async getContinuity(sectionId: string): Promise<{ section_id: string; continuity_metrics?: Record<string, number> }> {
     const { data } = await apiClient.get(`/api/blog/section/${encodeURIComponent(sectionId)}/continuity`);
     return data;
   },
 
-  async generateOutline(payload: { research: BlogResearchResponse; persona?: PersonaInfo; word_count?: number; custom_instructions?: string }): Promise<BlogOutlineResponse> {
-    // Use the direct outline generation endpoint
-    const { data } = await apiClient.post("/api/blog/outline/generate", payload);
-    return data;
-  },
 
   async refineOutline(payload: { outline: BlogOutlineSection[]; operation: string; section_id?: string; payload?: any }): Promise<BlogOutlineResponse> {
     const { data } = await apiClient.post("/api/blog/outline/refine", payload);