fix: react fixes

backend/app.py

@@ -33,7 +33,7 @@ app.add_middleware(
 
 sio = socketio.AsyncServer(
     cors_allowed_origins=CORS_ALLOWED_ORIGINS,
-    ping_timeout=120,
+    ping_timeout=1200,
     ping_interval=10,
     async_mode="asgi",
 )

backend/knet.py

@@ -138,11 +138,11 @@ class ResearchProgress:
         self.callback = callback
 
     async def update(self, progress: int, message: str):
-        self.progress = min(100, self.progress + progress)  # max 100
+        self.progress = int(min(100, self.progress + progress))  # max 100
         await self.callback({"progress": self.progress, "message": message})
 
     async def setter(self, progress: int, message: str):
-        self.progress = min(100, progress)  # max 100
+        self.progress = int(min(100, progress))  # max 100
         await self.callback({"progress": self.progress, "message": message})
 
 
@@ -189,21 +189,25 @@ class KNet:
             self.research_plan = self.generate_content(self.prompt.research_plan.format(topic=topic), schema=self.schema.research_plan)["steps"]
             self.logger.info(f"Research plan:\n{json.dumps(self.research_plan, indent=2)}")
 
-            # Generate initial search query
-            query = self.generate_content(
-                self.prompt.search_query.format(vertical=self.research_plan[self.idx_research_plan]), schema=self.schema.search_query
-            )["branches"][0]
-
-            # Initialize research tree
-            root_node = ResearchNode(query)
-            to_explore = deque([(root_node, 0)])  # (node, depth) pairs
-            explored_queries = set()  # {string, string, ...}
+            master_node = ResearchNode()
 
             await self.progress.update(0, "Starting research...")
 
             # Iterate on research plan
             for self.idx_research_plan, _ in enumerate(self.research_plan):
-                current_node, current_depth = to_explore.popleft()
+                # Generate initial search query
+                query = self.generate_content(
+                    self.prompt.search_query.format(
+                        vertical=self.research_plan[self.idx_research_plan], research_plan="None", past_queries="None", ctx_manager="None", n=1
+                    ),
+                    schema=self.schema.search_query,
+                )["branches"][0]
+
+                root_node = ResearchNode(query)
+                master_node.add_child(root_node)
+                to_explore = deque([(root_node, 0)])  # (node, depth) pairs
+                explored_queries = set()  # {string, string, ...}
+
                 await self.progress.update(100 / (len(self.research_plan) + 1), f"{self.research_plan[self.idx_research_plan]}")
 
                 while to_explore:
@@ -230,9 +234,9 @@ class KNet:
 
             # Generate final report
             await self.progress.update(100 / (len(self.research_plan) + 1), "Generating final report...")
-            final_report = self._generate_final_report(root_node, topic)
+            final_report = self._generate_final_report(master_node, topic)
 
-            self.logger.info(f"Research completed. Explored {len(explored_queries)} queries across {root_node.max_depth()} levels")
+            self.logger.info(f"Research completed. Explored {len(explored_queries)} queries across {master_node.max_depth()} levels")
             await self.progress.update(100, "Research complete!")
 
             with open("output.log.json", "w", encoding="utf-8") as f:
@@ -256,7 +260,7 @@ class KNet:
             report = []
             # Fill in report outline
             for i, heading in enumerate(outline["headings"]):
-                self.progress.update(100 / (len(outline["headings"] + 1)), "Generating report...")
+                self.progress.update(100 / (len(outline["headings"]) + 1), "Generating report...")
                 content = self.generate_content(
                     self.prompt.report_fillin.format(
                         topic=topic,
@@ -267,6 +271,8 @@ class KNet:
                     schema=self.schema.report_fillin,
                 )["content"]
                 report.append({"heading": heading, "content": content})
+            # Rasterize report
+            raster_report = f"# {outline['title']}\n\n" + "\n\n".join([f"## {r['heading']}\n\n{r['content']}" for r in report])
 
             # Collate multimedia content
             media_content = {"images": [], "videos": [], "links": [], "references": []}
@@ -297,9 +303,9 @@ class KNet:
                 }
 
             return {
-                "topic": root_node.query,
+                "topic": topic,
                 "timestamp": datetime.now().isoformat(),
-                "content": report,
+                "content": raster_report,
                 "media": media_content,
                 "research_tree": build_tree_structure(root_node),
                 "metadata": {
@@ -337,7 +343,7 @@ class KNet:
             # node -|-> child
             #       |-> child
             new_nodes = []
-            for branch in response.get("branches", []):
+            for branch in response.get("branches", [])[:1]:
                 child_node = node.add_child(branch)
                 new_nodes.append(child_node)
 
@@ -360,7 +366,7 @@ class KNet:
             if node.data:
                 findings = ("\n" + "-" * 10 + "Next data" + "-" * 10 + "\n").join([json.dumps(d, indent=2) for d in node.data])
                 response = self.generate_content(self.prompt.site_summary.format(query=node.query, findings=findings), temp=0.2)
-                self.ctx_manager.append(response)
+                self.ctx_manager.append(response) if isinstance(response, str) else None
 
             # Research manager takes decision to proceed or not
             prompt = self.prompt.continue_branch.format(

backend/research_node.py

@@ -3,7 +3,7 @@ from typing import Any, Dict, List, Optional
 
 
 class ResearchNode:
-    def __init__(self, query: str, parent: Optional["ResearchNode"] = None, depth: int = 0):
+    def __init__(self, query: str = "_", parent: Optional["ResearchNode"] = None, depth: int = 0):
         self.query = query
         self.parent = parent
         self.depth = depth
@@ -42,3 +42,11 @@ class ResearchNode:
         for child in self.children:
             data.extend(child.get_all_data())
         return data
+
+    def __repr__(self) -> dict:
+        return {
+            "query": self.query,
+            "depth": self.depth,
+            "children": [child.__repr__() for child in self.children],
+            "data": self.data,
+        }

frontend/src/components/ChatInterface.tsx

@@ -45,7 +45,6 @@ const ChatInterface = () => {
     sources: true,
     citations: false,
     max_depth: 1,
-    max_breadth: 3,
     num_sites_per_query: 5,
   });
 
@@ -131,11 +130,25 @@ const ChatInterface = () => {
           },
         ];
 
-        return {
+        // Save updated messages to localStorage
+        const updatedState = {
           ...prevState,
           isLoading: false,
           messages: newMessages,
         };
+
+        // Update localStorage with the new messages
+        const updatedData: ChatData = {
+          conversations: conversations.map((conv) => ({
+            ...conv,
+            messages: conv.id === currentConversationId ? newMessages : conv.messages || [],
+            lastUpdated: conv.id === currentConversationId ? new Date().toISOString() : conv.lastUpdated,
+          })),
+          currentConversationId,
+        };
+        saveToStorage(updatedData);
+
+        return updatedState;
       });
     });
 
@@ -230,7 +243,6 @@ const ChatInterface = () => {
       socket.emit("start_research", {
         topic: content,
         max_depth: researchOptions.max_depth,
-        max_breadth: researchOptions.max_breadth,
         num_sites_per_query: researchOptions.num_sites_per_query,
       });
     } catch (error) {

frontend/src/components/ResearchControls.tsx

@@ -65,17 +65,6 @@ const ResearchControls: React.FC<ResearchControlsProps> = ({ options, onOptionCh
           />
         </div>
 
-        <div className="space-y-2">
-          <Label htmlFor="max-breadth">Max Breadth</Label>
-          <Input
-            type="number"
-            id="max-breadth"
-            value={options.max_breadth}
-            onChange={(e) => onOptionChange({ ...options, max_breadth: parseInt(e.target.value, 10) })}
-            className="w-full"
-          />
-        </div>
-
         <div className="space-y-2">
           <Label htmlFor="num-sites-per-query">Number of Sites per Query</Label>
           <Input

frontend/src/components/ui/ChatLayout.tsx

@@ -2,12 +2,12 @@ import { Button } from "@/components/ui/button";
 import { Card } from "@/components/ui/card";
 import { ResizableHandle, ResizablePanel, ResizablePanelGroup } from "@/components/ui/resizable";
 import { ScrollArea } from "@/components/ui/scroll-area";
-import { Sheet, SheetContent, SheetTrigger } from "@/components/ui/sheet";
 import { Tabs, TabsContent, TabsList, TabsTrigger } from "@/components/ui/tabs";
 import { LayoutGrid, MessageCircle, Settings } from "lucide-react";
 import React from "react";
 import { ThemeToggle } from "./ThemeToggle";
 import Link from "next/link";
+import { Dialog, DialogContent, DialogDescription, DialogHeader, DialogTitle, DialogTrigger } from "@/components/ui/dialog";
 
 interface ChatLayoutProps {
   sidebar: React.ReactNode;
@@ -24,24 +24,27 @@ const ChatLayout: React.FC<ChatLayoutProps> = ({ sidebar, mainContent, settingsP
         </Link>
         <div className="flex-1" />
         <ThemeToggle />
-        <Sheet>
-          <SheetTrigger asChild>
-            <Button variant="ghost" size="icon">
+        <Dialog>
+          <DialogTrigger asChild>
+            <Button variant="ghost" size="icon" className="h-9 w-9" title="Settings">
               <Settings size={20} />
             </Button>
-          </SheetTrigger>
-          <SheetContent>
-            <div className="py-4">
-              <h2 className="text-lg font-semibold mb-4">Research Settings</h2>
-              {settingsPanel}
-            </div>
-          </SheetContent>
-        </Sheet>
+          </DialogTrigger>
+          <DialogContent>
+            <DialogHeader>
+              <DialogTitle>Research Settings</DialogTitle>
+              <DialogDescription>
+                Configure your research parameters and preferences.
+              </DialogDescription>
+            </DialogHeader>
+            {settingsPanel}
+          </DialogContent>
+        </Dialog>
       </header>
 
       <div className="flex-1 overflow-hidden">
         <ResizablePanelGroup direction="horizontal">
-          <ResizablePanel defaultSize={20} minSize={15} maxSize={30} className="hidden md:block">
+          <ResizablePanel defaultSize={25} minSize={17} maxSize={30} className="hidden md:block">
             <Card className="h-full rounded-none border-r border-t-0 border-l-0 border-b-0">
               <ScrollArea className="h-full">{sidebar}</ScrollArea>
             </Card>
@@ -49,7 +52,7 @@ const ChatLayout: React.FC<ChatLayoutProps> = ({ sidebar, mainContent, settingsP
 
           <ResizableHandle withHandle className="hidden md:flex" />
 
-          <ResizablePanel defaultSize={80}>
+          <ResizablePanel defaultSize={75}>
             <Tabs defaultValue="chat" className="h-full flex flex-col">
               <div className="p-4">
                 <TabsList className="">

frontend/src/lib/types.ts

@@ -16,7 +16,6 @@ export interface ResearchOptions {
   sources: boolean;
   citations: boolean;
   max_depth: number;
-  max_breadth: number;
   num_sites_per_query: number;
 }