Update main.py

main.py

@@ -159,7 +159,7 @@ async def __parse_paragraphs (content: str, batching: bool = False):
                 header=_n
                 h_cnt = __h_cnt
                 # print(_n, len(_n))
-            elif (len(_n.replace('#',''))): 
+            elif (len(_n.replace('#','').strip())): 
                 # remove accidental /n's in converted HTML content
 
                 if (batching and len(_p_json) >= 1):
@@ -171,7 +171,8 @@ async def __parse_paragraphs (content: str, batching: bool = False):
                         continue
 
                 _p_json.append({'header': header, 'h_cnt': h_cnt, 'content': _n.replace('\n',' '), 'count': 1})
-                
+
+            
 
         except:
             continue
@@ -203,7 +204,10 @@ async def __query_ml_predict(qtype: QType, content: str, header: str, token_limi
                 )
             
             print(time() - stopwatch)
-            if _r.status_code != 200: app_logger.info(_r.json())
+            if _r.status_code != 200: 
+                app_logger.info(_r.json())
+                return {"content": "", "style": QType.WH}
+                
             return {"content": _r.json()['choices'][0]['message']['content'], "style": QType.WH}
         
         case QType.STMT:
@@ -223,7 +227,10 @@ async def __query_ml_predict(qtype: QType, content: str, header: str, token_limi
                     }),
                     timeout=None
                 )
-            if _r.status_code != 200: app_logger.info(_r.json())
+            if _r.status_code != 200: 
+                app_logger.info(_r.json())
+                return {"content": "", "style": QType.STMT}
+                
             _r_content = _r.json()['choices'][0]['message']['content'].split('\n\n',1)[1]
             async with httpx.AsyncClient() as client:
                 _w = await client.post(