import json import os #import replicate import requests from PyQt5 import QtWidgets, QtGui, QtCore # from dotenv import load_dotenv import os # load_dotenv() # 加载 .env 文件中的环境变量 url = 'http://43.139.183.222:5000' class ImportPDFDialog6(QtWidgets.QWidget): def __init__(self): super().__init__() self.initUI() self.processed_image_path = None # 存储处理后的图片路径 def initUI(self): self.setWindowTitle('图像处理参数输入') # 布局 layout = QtWidgets.QVBoxLayout() # 文件选择行 self.file_input = QtWidgets.QLineEdit(self) # self.file_input.setPlaceholderText("选择文件路径 (如 1.jpg)") layout.addWidget(self.file_input) self.browse_button = QtWidgets.QPushButton('浏览', self) self.browse_button.clicked.connect(self.browse_file) layout.addWidget(self.browse_button) self.prompt_input = QtWidgets.QLineEdit(self) self.prompt_input.setPlaceholderText("提示词 ") layout.addWidget(self.prompt_input) self.creativity_input = QtWidgets.QDoubleSpinBox(self) self.creativity_input.setRange(0, 1) self.creativity_input.setSingleStep(0.01) self.creativity_input.setValue(0.35) # 默认值 layout.addWidget(QtWidgets.QLabel("创造力 (0 - 3):")) layout.addWidget(self.creativity_input) self.resemblance_input = QtWidgets.QDoubleSpinBox(self) self.resemblance_input.setRange(0, 3) self.resemblance_input.setSingleStep(0.01) self.resemblance_input.setValue(0.6) # 默认值 layout.addWidget(QtWidgets.QLabel("相似度 (0 - 3):")) layout.addWidget(self.resemblance_input) self.scale_factor_input = QtWidgets.QSpinBox(self) self.scale_factor_input.setRange(2, 8) self.scale_factor_input.setValue(2) # 默认值 layout.addWidget(QtWidgets.QLabel("放大倍数:")) layout.addWidget(self.scale_factor_input) self.submit_button = QtWidgets.QPushButton('提交', self) self.submit_button.clicked.connect(self.submit) layout.addWidget(self.submit_button) # 合并图片显示和处理结果区域 self.output_area = QtWidgets.QVBoxLayout() self.image_label = QtWidgets.QLabel(self) # self.image_label.setText("加载的图片将显示在这里") self.image_label.setAlignment(QtCore.Qt.AlignCenter) self.output_area.addWidget(self.image_label) self.result_display = QtWidgets.QTextEdit(self) self.result_display.setReadOnly(True) self.output_area.addWidget(QtWidgets.QLabel("处理结果:")) self.output_area.addWidget(self.result_display) layout.addLayout(self.output_area) # 下载按钮 self.download_button = QtWidgets.QPushButton('下载', self) self.download_button.clicked.connect(self.download_image) self.download_button.setEnabled(False) # 初始不可用 layout.addWidget(self.download_button) self.setLayout(layout) def browse_file(self): options = QtWidgets.QFileDialog.Options() file_name, _ = QtWidgets.QFileDialog.getOpenFileName(self, "选择图像文件", "", "Image Files (*.png *.jpg *.jpeg *.bmp)", options=options) if file_name: self.file_input.setText(file_name) self.display_image(file_name) # 显示选中的图片 def display_image(self, file_path): pixmap = QtGui.QPixmap(file_path) self.image_label.setPixmap(pixmap.scaled(300, 300, QtCore.Qt.KeepAspectRatio)) def upload_image(self, url, file_path, json_data): imgurl = url + "/upload" with open(file_path, 'rb') as file: files = {'file': file} data = {'data': json.dumps(json_data)} # 将 JSON 数据转换为字符串 response = requests.post(imgurl, files=files, data=data) print(response.status_code) response_json = response.json() # 先解析 JSON 响应 print(response_json) # 打印响应内容以检查其格式 # 确保响应是一个列表，并且至少有一个元素 if isinstance(response_json, list) and len(response_json) > 0: self.processed_image_path = response_json[0] # 从列表中获取 URL else: # 处理不符合预期的情况 print("Unexpected response format:", response_json) return self.result_display.setPlainText(str(response_json)) # 显示处理结果 self.download_button.setEnabled(True) # 启用下载按钮 self.display_processed_image() # 显示处理后的图片 def submit(self): try: creativity = self.creativity_input.value() file_path = self.file_input.text() urlprompt = self.prompt_input.text() + " best quality, highres, " prompt = urlprompt resemblance = self.resemblance_input.value() scale_factor = self.scale_factor_input.value() if not os.path.exists(file_path): raise ValueError("文件不存在，请选择有效的文件。") input_data = { "seed": 1337, "prompt": prompt, "dynamic": 6, "handfix": "disabled", "pattern": False, "sharpen": 0, "sd_model": "juggernaut_reborn.safetensors [338b85bc4f]", "scheduler": "DPM++ 3M SDE Karras", "creativity": creativity, "lora_links": "", "downscaling": False, "resemblance": resemblance, "scale_factor": scale_factor, "tiling_width": 112, "tiling_height": 144, "output_format": "png", "custom_sd_model": "", "negative_prompt": "(worst quality, low quality, normal quality:2) JuggernautNegative-neg", "num_inference_steps": 18, "downscaling_resolution": 768 } # 确保调用时只传递三个参数 self.upload_image(url, file_path, input_data) except Exception as e: QtWidgets.QMessageBox.critical(self, "错误", str(e)) def run(url): newurl = url + "/data" data = {'key': 'value'} # 你要发送的数据 response = requests.post(newurl, json=data) print(response.status_code) # 打印状态码 print(response.json()) # 打印返回的 JSON 数据 def run2(url): newurl = url + "/dataimg" data = {'key': 'value'} # 你要发送的数据 response = requests.post(newurl, json=data) print(response.status_code) # 打印状态码 print(response.json()) # 打印返回的 JSON 数据 # if __name__ == '__main__': # upload_image(url, '1.jpg', {'key': 'value'}) def display_processed_image(self): # 下载并显示处理后的图片 response = requests.get(self.processed_image_path) if response.status_code == 200: pixmap = QtGui.QPixmap() pixmap.loadFromData(response.content) self.image_label.setPixmap(pixmap.scaled(300, 300, QtCore.Qt.KeepAspectRatio)) def download_image(self): if self.processed_image_path: image_url = self.processed_image_path file_name, _ = QtWidgets.QFileDialog.getSaveFileName(self, "保存处理后的图片", "", "Image Files (*.png *.jpg *.jpeg)") if file_name: try: response = requests.get(image_url) response.raise_for_status() # 检查请求是否成功 with open(file_name, 'wb') as f: f.write(response.content) QtWidgets.QMessageBox.information(self, "成功", "图片下载完成！") except requests.exceptions.RequestException as e: QtWidgets.QMessageBox.critical(self, "错误", f"下载失败：{e}") except Exception as e: QtWidgets.QMessageBox.critical(self, "错误", f"文件保存失败：{e}") if __name__ == '__main__': import sys app = QtWidgets.QApplication(sys.argv) # 创建并显示登录界面 # login_dialog = LoginDialog() # if login_dialog.exec_() == QtWidgets.QDialog.Accepted: # 只有在登录成功后才执行以下代码 ex = ImportPDFDialog6() ex.show() sys.exit(app.exec_())