Update app.py

app.py

@@ -18,7 +18,6 @@ def detect_language_script(text: str) -> tuple[str, str]:
         # Use confidence threshold to avoid false detections
         lang_detect = langdetect.detect_langs(text)
         if lang_detect[0].prob > 0.8:
-            # Only accept high confidence detections
             lang = lang_detect[0].lang
         else:
             lang = 'en'  # Default to English if unsure
@@ -35,7 +34,6 @@ def detect_language_script(text: str) -> tuple[str, str]:
 def is_romanized_indic(text: str) -> bool:
     """Check if text appears to be romanized Indic language.
     More strict pattern matching."""
-    # Common Bengali romanized patterns with word boundaries
     bengali_patterns = [
         r'\b(ami|tumi|apni)\b',  # Common pronouns
         r'\b(ache|achen|thako|thaken)\b',  # Common verbs
@@ -43,21 +41,18 @@ def is_romanized_indic(text: str) -> bool:
         r'\b(ki|kothay|keno)\b'  # Common question words
     ]
     
-    # Require multiple matches to confirm it's actually Bengali
     text_lower = text.lower()
     matches = sum(1 for pattern in bengali_patterns if re.search(pattern, text_lower))
     return matches >= 2  # Require at least 2 matches to consider it Bengali
 
 def translate_text(text: str, target_lang='en') -> tuple[str, str, bool]:
     """Translate text to target language, with more conservative translation logic."""
-    # Skip translation for very short inputs or basic greetings
     if len(text.split()) <= 2 or text.lower() in ['hello', 'hi', 'hey']:
         return text, 'en', False
     
     original_lang, script = detect_language_script(text)
     is_transliterated = False
     
-    # Only process if confident it's non-English
     if original_lang != 'en' and len(text.split()) > 2:
         try:
             translator = GoogleTranslator(source='auto', target=target_lang)
@@ -67,7 +62,6 @@ def translate_text(text: str, target_lang='en') -> tuple[str, str, bool]:
             print(f"Translation error: {e}")
             return text, 'en', False
     
-    # Check for romanized Indic text only if it's a longer input
     if original_lang == 'en' and len(text.split()) > 2 and is_romanized_indic(text):
         text = romanized_to_bengali(text)
         return translate_text(text, target_lang)  # Recursive call with Bengali script
@@ -117,17 +111,19 @@ def generate_image(prompt: str) -> str:
         response = image_client.text_to_image(
             prompt,
             parameters={
-                "negative_prompt": "blurry, bad quality, nsfw",
+                "negative_prompt": "(worst quality, low quality, illustration, 3d, 2d, painting, cartoons, sketch), open mouth",
                 "num_inference_steps": 30,
-                "guidance_scale": 7.5
+                "guidance_scale": 7.5,
+                "sampling_steps": 15,  # Adjusted parameter
+                "upscaler": "4x-UltraSharp",
+                "denoising_strength": 0.5,  # Denoising strength between 0.1 and 0.5
             }
         )
-        # Save the image and return the path or base64 string
-        # Note: Implementation depends on how you want to handle the image output
-        return response
+        return response  # Assuming response contains the image as required
     except Exception as e:
         print(f"Image generation error: {e}")
         return None
+
 def romanized_to_bengali(text: str) -> str:
     """Convert romanized Bengali text to Bengali script."""
     bengali_mappings = {
@@ -171,20 +167,17 @@ def respond(
         return
 
     # Check if this is an image generation request
+       """Handle user message and respond accordingly."""
+    # Check for image requests first
     if is_image_request(message):
         try:
             image = generate_image(message)
             if image:
-                yield f"Here's your generated image based on: {message}"
-                # You'll need to implement the actual image display logic
-                # depending on your Gradio interface requirements
-                return
+                return f"Here's your generated image based on: {message}"
             else:
-                yield "Sorry, I couldn't generate the image. Please try again."
-                return
+                return "Sorry, I couldn't generate the image. Please try again."
         except Exception as e:
-            yield f"An error occurred while generating the image: {str(e)}"
-            return
+            return f"An error occurred while generating the image: {str(e)}"
 
     # Handle translation with more conservative approach
     translated_msg, original_lang, was_transliterated = translate_text(message)