Add files via upload
This commit is contained in:
@@ -0,0 +1,156 @@
|
||||
#!/usr/bin/env python3
|
||||
import sys
|
||||
import gi
|
||||
import subprocess
|
||||
import threading
|
||||
|
||||
gi.require_version("Gtk", "4.0")
|
||||
from gi.repository import Gtk, GLib
|
||||
|
||||
AI_SCRIPT = "/home/bob/ai_capture/ai.py"
|
||||
|
||||
MODELS = [
|
||||
"qwen/qwen-2.5-vl-7b-instruct:free",
|
||||
"meta-llama/llama-3.2-11b-vision-instruct",
|
||||
"google/gemini-2.5-flash-image",
|
||||
]
|
||||
|
||||
class App(Gtk.Application):
|
||||
def __init__(self, image_path):
|
||||
super().__init__(application_id="ai.capture.ui")
|
||||
self.image_path = image_path
|
||||
|
||||
def do_activate(self):
|
||||
win = Gtk.ApplicationWindow(application=self)
|
||||
win.set_title("AI Capture")
|
||||
win.set_default_size(900, 800)
|
||||
|
||||
root = Gtk.Box(orientation=Gtk.Orientation.VERTICAL, spacing=8)
|
||||
root.set_margin_top(10)
|
||||
root.set_margin_bottom(10)
|
||||
root.set_margin_start(10)
|
||||
root.set_margin_end(10)
|
||||
|
||||
# === IMAGE ===
|
||||
picture = Gtk.Picture.new_for_filename(self.image_path)
|
||||
picture.set_hexpand(True)
|
||||
picture.set_vexpand(True)
|
||||
picture.set_content_fit(Gtk.ContentFit.CONTAIN)
|
||||
|
||||
frame = Gtk.Frame()
|
||||
frame.set_hexpand(True)
|
||||
frame.set_vexpand(True)
|
||||
frame.set_child(picture)
|
||||
root.append(frame)
|
||||
|
||||
# === INPUT ROW ===
|
||||
row = Gtk.Box(orientation=Gtk.Orientation.HORIZONTAL, spacing=6)
|
||||
|
||||
self.entry = Gtk.Entry()
|
||||
self.entry.set_placeholder_text("Bu görüntü hakkında sorunuzu yazın…")
|
||||
self.entry.set_hexpand(True)
|
||||
row.append(self.entry)
|
||||
|
||||
self.ask_btn = Gtk.Button(label="Sor")
|
||||
self.ask_btn.connect("clicked", self.on_ask)
|
||||
row.append(self.ask_btn)
|
||||
|
||||
menu_btn = Gtk.Button(label="⋮")
|
||||
row.append(menu_btn)
|
||||
|
||||
root.append(row)
|
||||
|
||||
# === OUTPUT ===
|
||||
self.output = Gtk.TextView()
|
||||
self.output.set_editable(False)
|
||||
self.output.set_wrap_mode(Gtk.WrapMode.WORD_CHAR)
|
||||
|
||||
out_scroll = Gtk.ScrolledWindow()
|
||||
out_scroll.set_vexpand(True)
|
||||
out_scroll.set_policy(Gtk.PolicyType.NEVER, Gtk.PolicyType.AUTOMATIC)
|
||||
out_scroll.set_child(self.output)
|
||||
|
||||
root.append(out_scroll)
|
||||
|
||||
# === MODEL POPOVER ===
|
||||
popover = Gtk.Popover()
|
||||
box = Gtk.Box(orientation=Gtk.Orientation.VERTICAL, spacing=4)
|
||||
box.set_margin_top(8)
|
||||
box.set_margin_bottom(8)
|
||||
box.set_margin_start(8)
|
||||
box.set_margin_end(8)
|
||||
|
||||
self.checks = {}
|
||||
for m in MODELS:
|
||||
cb = Gtk.CheckButton(label=m)
|
||||
cb.set_active(m == MODELS[0])
|
||||
self.checks[m] = cb
|
||||
box.append(cb)
|
||||
|
||||
popover.set_child(box)
|
||||
popover.set_parent(menu_btn)
|
||||
menu_btn.connect("clicked", lambda *_: popover.popup())
|
||||
|
||||
win.set_child(root)
|
||||
win.present()
|
||||
|
||||
def on_ask(self, _):
|
||||
prompt = self.entry.get_text().strip()
|
||||
if not prompt:
|
||||
return
|
||||
|
||||
models = [m for m, cb in self.checks.items() if cb.get_active()]
|
||||
if not models:
|
||||
self.set_output("❌ En az bir model seçmelisiniz.")
|
||||
return
|
||||
|
||||
self.ask_btn.set_sensitive(False)
|
||||
self.set_output("⏳ Yanıt alınıyor…")
|
||||
|
||||
threading.Thread(
|
||||
target=self.run_ai,
|
||||
args=(prompt, models),
|
||||
daemon=True
|
||||
).start()
|
||||
|
||||
def run_ai(self, prompt, models):
|
||||
for model in models:
|
||||
try:
|
||||
result = subprocess.run(
|
||||
[
|
||||
"python",
|
||||
AI_SCRIPT,
|
||||
self.image_path,
|
||||
prompt,
|
||||
model
|
||||
],
|
||||
capture_output=True,
|
||||
text=True,
|
||||
timeout=120
|
||||
)
|
||||
|
||||
if result.returncode == 0 and result.stdout.strip():
|
||||
GLib.idle_add(self.on_done, result.stdout.strip())
|
||||
return
|
||||
except Exception as e:
|
||||
last_error = str(e)
|
||||
|
||||
GLib.idle_add(self.on_done, f"❌ Tüm modeller başarısız.\n{last_error}")
|
||||
|
||||
def on_done(self, text):
|
||||
self.set_output(text)
|
||||
self.ask_btn.set_sensitive(True)
|
||||
return False
|
||||
|
||||
def set_output(self, text):
|
||||
buf = self.output.get_buffer()
|
||||
buf.set_text(text)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
if len(sys.argv) != 2:
|
||||
print("Kullanım: ui.py <image_path>")
|
||||
sys.exit(1)
|
||||
|
||||
app = App(sys.argv[1])
|
||||
app.run()
|
||||
Reference in New Issue
Block a user