Update app.py

app.py

@@ -41,18 +41,24 @@ async def root():
     return {"message": "Patent Analyzer API is running"}
 
 @app.post("/search")
-async def search(request: SearchRequest):
-    if not request.query:
+async def search(query: str, data_type: str = None):
+    if not query:
         raise HTTPException(status_code=400, detail="No query provided")
     
     try:
-        # Remove filetype:pdf if present since DDGS might handle it differently
-        clean_query = request.query.replace('filetype:pdf', '').strip()
-        results = search_web(clean_query, max_references=5)
+        if data_type == "pdf" or data_type is None:
+            search_query = f"{query} filetype:pdf"
+        elif data_type == "patent":
+            search_query = f"{query} site:patents.google.com"
+        else:
+            search_query = query
+        
+        results = search_web(search_query, max_references=5)
         return {"results": results}
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Error performing search: {str(e)}")
 
+
 @app.post("/analyze")
 async def analyze(request: AnalysisRequest):
     if not request.patent_background or not request.pdf_url: