ai: Press F for respect. :(

jarvis.py

@@ -128,7 +128,7 @@ def extract_docx_content(fp):
                         ocr_text = pytesseract.image_to_string(img)
                         if ocr_text.strip():
                             content += ocr_text + "\n"
-                    except Exception:
+                    except:
                         pass
     except Exception as e:
         content += f"{fp}: {e}"
@@ -151,7 +151,7 @@ def extract_excel_content(fp):
                         ocr_text = pytesseract.image_to_string(pil_img)
                         if ocr_text.strip():
                             content += ocr_text + "\n"
-                    except Exception:
+                    except:
                         pass
     except Exception as e:
         content += f"{fp}: {e}"
@@ -171,15 +171,14 @@ def extract_pptx_content(fp):
                         ocr_text = pytesseract.image_to_string(img)
                         if ocr_text.strip():
                             content += ocr_text + "\n"
-                    except Exception:
+                    except:
                         pass
-            if slide.shapes:
-                for shape in slide.shapes:
-                    if shape.has_table:
-                        table = shape.table
-                        for row in table.rows:
-                            cells = [cell.text for cell in row.cells]
-                            content += "\t".join(cells) + "\n"
+            for shape in slide.shapes:
+                if shape.has_table:
+                    table = shape.table
+                    for row in table.rows:
+                        cells = [cell.text for cell in row.cells]
+                        content += "\t".join(cells) + "\n"
     except Exception as e:
         content += f"{fp}: {e}"
     return content.strip()
@@ -236,10 +235,7 @@ async def chat_with_model_async(history, user_input, model_display, sess, custom
     model_key = get_model_key(model_display)
     cfg = MODEL_CONFIG.get(model_key, DEFAULT_CONFIG)
     msgs = [{"role": "user", "content": u} for u, _ in history] + [{"role": "assistant", "content": a} for _, a in history if a]
-    if model_key == DEFAULT_MODEL_KEY and INTERNAL_TRAINING_DATA:
-        prompt = INTERNAL_TRAINING_DATA
-    else:
-        prompt = custom_prompt or SYSTEM_PROMPT_MAPPING.get(model_key, SYSTEM_PROMPT_DEFAULT)
+    prompt = INTERNAL_TRAINING_DATA if model_key == DEFAULT_MODEL_KEY and INTERNAL_TRAINING_DATA else (custom_prompt or SYSTEM_PROMPT_MAPPING.get(model_key, SYSTEM_PROMPT_DEFAULT))
     msgs.insert(0, {"role": "system", "content": prompt})
     msgs.append({"role": "user", "content": user_input})
     if sess.active_candidate:
@@ -285,7 +281,15 @@ async def respond_async(multi, history, model_display, sess, custom_prompt):
         inp += msg_input["text"]
     history.append([inp, RESPONSES["RESPONSE_8"]])
     yield history, gr.update(interactive=False, submit_btn=False, stop_btn=True), sess
-    ai = await chat_with_model_async(history, inp, model_display, sess, custom_prompt)
+    task = asyncio.create_task(chat_with_model_async(history, inp, model_display, sess, custom_prompt))
+    done, pending = await asyncio.wait({task, sess.stop_event.wait()}, return_when=asyncio.FIRST_COMPLETED)
+    if sess.stop_event.is_set():
+        task.cancel()
+        history[-1][1] = RESPONSES["RESPONSE_1"]
+        yield history, gr.update(value="", interactive=True, submit_btn=True, stop_btn=False), sess
+        sess.stop_event.clear()
+        return
+    ai = task.result()
     history[-1][1] = ""
     buffer = []
     last_update = asyncio.get_event_loop().time()
@@ -297,12 +301,12 @@ async def respond_async(multi, history, model_display, sess, custom_prompt):
             return
         buffer.append(char)
         current_time = asyncio.get_event_loop().time()
-        if len(buffer) >= 8 or (current_time - last_update) > 0.04:
+        if len(buffer) >= 4 or (current_time - last_update) > 0.02:
             history[-1][1] += "".join(buffer)
             buffer.clear()
             last_update = current_time
             yield history, gr.update(interactive=False, submit_btn=False, stop_btn=True), sess
-            await asyncio.sleep(0.020)
+            await asyncio.sleep(0.012)
     if buffer:
         history[-1][1] += "".join(buffer)
         yield history, gr.update(interactive=False, submit_btn=False, stop_btn=True), sess