Update app.py

app.py

@@ -1456,23 +1456,27 @@ def handle_generation(h,w,d):
     if not working:
 
         working = True
+
+        if len(d) > 0:
+            d = re.sub(r",( ){1,}",". ",d)
+            d_lines = re.split(r"([\n]){1,}", d)
+            
+            for line_index in range(len(d_lines)):
+                d_lines[line_index] = d_lines[line_index].strip()
+                if re.sub(r'[\.]$', '', d_lines[line_index]) == d_lines[line_index]:
+                    d_lines[line_index] = d_lines[line_index].strip() + "."
+            d = " ".join(d_lines)
         
-        d = re.sub(r",( ){1,}",". ",d)
-        d_lines = re.split(r"([\n]){1,}", d)
-        
-        for line_index in range(len(d_lines)):
-            d_lines[line_index] = d_lines[line_index].strip()
-            if re.sub(r'[\.]$', '', d_lines[line_index]) == d_lines[line_index]:
-                d_lines[line_index] = d_lines[line_index].strip() + "."
-        d = " ".join(d_lines)
-    
-        pos_d = re.sub(r"([ \t]){1,}", " ", d).lower().strip()
-        pos_d = pos_d if pos_d == "" else summarize(translate(pos_d))
-        pos_d = re.sub(r"([ \t]){1,}", " ", pos_d).lower().strip()
+            d = re.sub(r"([ \t]){1,}", " ", d).lower().strip()
+            if len(d) > 400:
+                d = pos_d if pos_d == "" else summarize(translate(d))
+            else:
+                d = pos_d if pos_d == "" else translate(d)
+            d = re.sub(r"([ \t]){1,}", " ", d).lower().strip()
     
         neg = f"Textual, Text, Distorted, Fake, Discontinuous, Blurry, Doll-Like, Overly Plastic, Low Quality, Paint, Smoothed, Artificial, Phony, Gaudy, Digital Effects."
         q = "\""
-        pos = f'HQ Hyper-realistic professional photograph{ pos_d if pos_d == "" else ": " + pos_d }.'
+        pos = f'HQ Hyper-realistic professional photograph{ d if d == "" else ": " + d }.'
     
         print(f"""
             Positive: {pos}