Update main.py

main.py

@@ -39,11 +39,31 @@ rate_limit_store = defaultdict(lambda: {"count": 0, "timestamp": time.time()})
 CLEANUP_INTERVAL = 60  # seconds
 RATE_LIMIT_WINDOW = 60  # seconds
 
-# Define ImageResponse if needed
+# Define ImageResponseModel
 class ImageResponseModel(BaseModel):
     images: str
     alt: str
 
+def strip_markdown(text: str) -> str:
+    """
+    Strips markdown syntax from the given text to ensure plain text.
+    """
+    # Remove bold (**text** or __text__)
+    text = re.sub(r'(\*\*|__)(.*?)\1', r'\2', text)
+    # Remove italic (*text* or _text_)
+    text = re.sub(r'(\*|_)(.*?)\1', r'\2', text)
+    # Remove inline code (`code`)
+    text = re.sub(r'`(.*?)`', r'\1', text)
+    # Remove links [text](url)
+    text = re.sub(r'\[(.*?)\]\((.*?)\)', r'\1', text)
+    # Remove images ![alt](url)
+    text = re.sub(r'!\[(.*?)\]\((.*?)\)', r'\1', text)
+    # Remove headers (# Header)
+    text = re.sub(r'#+\s+(.*)', r'\1', text)
+    # Remove any remaining markdown characters
+    text = re.sub(r'[*_`>#]', '', text)
+    return text
+
 # Updated Blackbox Class
 class Blackbox:
     label = "Blackbox AI"
@@ -206,6 +226,8 @@ class Blackbox:
     def clean_response(text: str) -> str:
         pattern = r'^\$\@\$v=undefined-rv1\$\@\$'
         cleaned_text = re.sub(pattern, '', text)
+        # Strip markdown syntax to prevent bold text
+        cleaned_text = strip_markdown(cleaned_text)
         return cleaned_text
 
     @classmethod
@@ -633,9 +655,14 @@ async def chat_completions(request: ChatRequest, req: Request, api_key: str = De
                     websearch=False  # Modify if websearch is needed
                 ):
                     if isinstance(chunk, ImageResponseModel):
+                        # Yield image URLs as plain text
                         yield f"Image URL: {chunk.images}\n"
                     else:
-                        yield f"{chunk}\n"
+                        # Ensure chunk is a string and yield as plain text
+                        if isinstance(chunk, str):
+                            yield f"{chunk}\n"
+                        else:
+                            yield f"{str(chunk)}\n"
 
             logger.info(f"Initiating streaming response for API key: {api_key} | IP: {client_ip}")
             return StreamingResponse(content_generator(), media_type='text/plain')