Spaces:
Sleeping
Sleeping
Saqib commited on
Update modules/app.py
Browse files- modules/app.py +1 -40
modules/app.py
CHANGED
|
@@ -120,29 +120,6 @@ async def img2location(request: Request):
|
|
| 120 |
|
| 121 |
if not image_url:
|
| 122 |
raise HTTPException(status_code=400, detail="image_url not found in request")
|
| 123 |
-
|
| 124 |
-
def extract_coordinates(text):
|
| 125 |
-
# Split the text into lines
|
| 126 |
-
lines = text.split('\n')
|
| 127 |
-
|
| 128 |
-
# Iterate through each line to find the one that contains the coordinates
|
| 129 |
-
for line in lines:
|
| 130 |
-
try:
|
| 131 |
-
if line.startswith("Coordinates:"):
|
| 132 |
-
# Remove the label and split by comma
|
| 133 |
-
coords = line.replace("Coordinates:", "").strip()
|
| 134 |
-
lat, lon = coords.split(',')
|
| 135 |
-
|
| 136 |
-
# Further split by space to isolate numerical values
|
| 137 |
-
latitude = float(lat.split('°')[0].strip())
|
| 138 |
-
longitude = float(lon.split('°')[0].strip())
|
| 139 |
-
|
| 140 |
-
return latitude, longitude
|
| 141 |
-
except Exception as e:
|
| 142 |
-
print("Error:", e)
|
| 143 |
-
return None
|
| 144 |
-
# Return None if no coordinates are found
|
| 145 |
-
return None
|
| 146 |
|
| 147 |
headers = {
|
| 148 |
'accept': '*/*',
|
|
@@ -178,23 +155,7 @@ async def img2location(request: Request):
|
|
| 178 |
if response.status != 200:
|
| 179 |
return f"Failed to upload image: HTTP {response.status}"
|
| 180 |
json_response = await response.json()
|
| 181 |
-
|
| 182 |
-
if 'message' in json_response:
|
| 183 |
-
json_response['message'] = json_response['message'].strip()
|
| 184 |
-
coordinates = extract_coordinates(json_response['message'])
|
| 185 |
-
|
| 186 |
-
if coordinates:
|
| 187 |
-
latitude, longitude = coordinates
|
| 188 |
-
google_maps = f"https://www.google.com/maps/search/?api=1&query={latitude},{longitude}"
|
| 189 |
-
|
| 190 |
-
json_response['message'] += f"\n\nView on Google Maps: {google_maps}"
|
| 191 |
-
|
| 192 |
-
if img2location_name in json_response['message'].lower():
|
| 193 |
-
raise HTTPException(status_code=400, detail="We are not allowed to process this image. Please try another one.")
|
| 194 |
-
|
| 195 |
-
return json_response
|
| 196 |
-
|
| 197 |
-
raise ValueError(f"Unexpected response: {json_response}")
|
| 198 |
|
| 199 |
@app.post("/pixart-sigma")
|
| 200 |
async def pixart_sigma(request: Request):
|
|
|
|
| 120 |
|
| 121 |
if not image_url:
|
| 122 |
raise HTTPException(status_code=400, detail="image_url not found in request")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 123 |
|
| 124 |
headers = {
|
| 125 |
'accept': '*/*',
|
|
|
|
| 155 |
if response.status != 200:
|
| 156 |
return f"Failed to upload image: HTTP {response.status}"
|
| 157 |
json_response = await response.json()
|
| 158 |
+
return json_response
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 159 |
|
| 160 |
@app.post("/pixart-sigma")
|
| 161 |
async def pixart_sigma(request: Request):
|