init

app.py

@@ -4,6 +4,9 @@ import torch
 from transformers import Qwen2_5_VLForConditionalGeneration, AutoProcessor
 import os
 from datetime import datetime
+from PIL import Image
+import requests
+from io import BytesIO
 
 # Model setup
 processor = AutoProcessor.from_pretrained("deepguess/weather-vlm-qwen2.5-7b", trust_remote_code=True)
@@ -79,10 +82,39 @@ MODE_INFO = {
     "research": "In-depth research perspective"
 }
 
+# Example URLs for different weather data types
+EXAMPLE_URLS = {
+    "SPC Convective Outlook": "https://www.spc.noaa.gov/products/outlook/day1otlk.gif",
+    "SPC Mesoanalysis (Surface)": "https://inside.nssl.noaa.gov/ewp/wp-content/uploads/sites/22/2019/05/19ZSPCCape.png",
+    "College of DuPage Nexrad": "https://upload.wikimedia.org/wikipedia/commons/9/9b/NEXRAD_radar_of_an_EF2_tornado_in_Kansas_on_March_13%2C_2024.png",
+    "Tropical Tidbits GFS 500mb": "https://sites.gatech.edu/eas-mesoscale-blog/files/2023/04/Hebert_Fig4-768x531.png",
+    "GOES-16 Visible": "https://cdn.star.nesdis.noaa.gov/GOES16/ABI/CONUS/GEOCOLOR/1250x750.jpg",
+    "KOUN Sounding": "https://sharppy.github.io/SHARPpy/_images/gui.sharppy.png",
+}
+
+def load_image_from_url(url):
+    """Load an image from URL."""
+    try:
+        response = requests.get(url, timeout=10, headers={'User-Agent': 'Mozilla/5.0'})
+        response.raise_for_status()
+        img = Image.open(BytesIO(response.content))
+        # Convert to RGB if necessary
+        if img.mode != 'RGB':
+            img = img.convert('RGB')
+        return img
+    except Exception as e:
+        return None, f"Error loading image from URL: {str(e)}"
+
 @spaces.GPU(duration=90)
-def analyze_weather_image(image, analysis_type, custom_prompt, analysis_mode, temperature, max_tokens, top_p):
-    if image is None:
-        return "Please upload an image to analyze."
+def analyze_weather_image(image, image_url, analysis_type, custom_prompt, analysis_mode, temperature, max_tokens, top_p):
+    # Handle image input - either direct upload or URL
+    if image_url and image_url.strip():
+        result = load_image_from_url(image_url.strip())
+        if isinstance(result, tuple):  # Error case
+            return result[1]
+        image = result
+    elif image is None:
+        return "Please upload an image or provide an image URL to analyze."
     
     # Move model to GPU
     model.cuda()
@@ -124,7 +156,8 @@ def analyze_weather_image(image, analysis_type, custom_prompt, analysis_mode, te
     
     # Add metadata
     timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S UTC")
-    metadata = f"\n\n---\n*Analysis completed: {timestamp} | Mode: {analysis_mode} | Type: {analysis_type}*"
+    source = "URL" if image_url and image_url.strip() else "Upload"
+    metadata = f"\n\n---\n*Analysis completed: {timestamp} | Mode: {analysis_mode} | Type: {analysis_type} | Source: {source}*"
     
     return response + metadata
 
@@ -135,18 +168,35 @@ with gr.Blocks(title=TITLE, theme=gr.themes.Base(), css="""
     #analysis-output { font-family: 'Monaco', 'Menlo', monospace; font-size: 13px; }
     .gr-button-primary { background-color: #2563eb; }
     .gr-button-primary:hover { background-color: #1e40af; }
+    .url-button { min-width: 120px; }
 """) as demo:
     gr.Markdown(f"# {TITLE}")
     gr.Markdown(DESCRIPTION)
     
     with gr.Row():
         with gr.Column(scale=1):
-            # Image input
-            image_input = gr.Image(
-                label="Upload Weather Image",
-                type="pil",
-                elem_id="image-upload"
-            )
+            # Image input options
+            with gr.Tabs():
+                with gr.Tab("Upload Image"):
+                    image_input = gr.Image(
+                        label="Upload Weather Image",
+                        type="pil",
+                        elem_id="image-upload"
+                    )
+                
+                with gr.Tab("Image URL"):
+                    image_url_input = gr.Textbox(
+                        label="Image URL",
+                        placeholder="https://example.com/weather-image.jpg",
+                        lines=1
+                    )
+                    
+                    # Quick URL examples
+                    gr.Markdown("**Quick Examples:**")
+                    url_buttons = []
+                    for name, url in EXAMPLE_URLS.items():
+                        btn = gr.Button(name, size="sm", elem_classes="url-button")
+                        btn.click(lambda u=url: u, outputs=image_url_input)
             
             # Analysis type selector
             analysis_type = gr.Dropdown(
@@ -250,25 +300,6 @@ with gr.Blocks(title=TITLE, theme=gr.themes.Base(), css="""
                 outputs=custom_prompt
             )
     
-    # Examples section
-    example_data = [
-        ["examples/500mb_heights.jpg", "Model Output", "technical", 0.7, 512],
-        ["examples/sounding.jpg", "Sounding Analysis", "educational", 0.7, 512],
-        ["examples/radar_composite.jpg", "Radar/Satellite", "operational", 0.7, 384],
-        ["examples/spc_outlook.jpg", "Severe Weather", "operational", 0.7, 512],
-        ["examples/ensemble_spaghetti.jpg", "Ensemble Analysis", "research", 0.8, 640],
-    ] if os.path.exists("examples") else []
-    
-    if example_data:
-        gr.Examples(
-            examples=example_data,
-            inputs=[image_input, analysis_type, analysis_mode, temperature, max_tokens],
-            outputs=output,
-            fn=lambda img, typ, mode, temp, tok: analyze_weather_image(img, typ, "", mode, temp, tok, 0.95),
-            cache_examples=False,
-            label="Example Analyses"
-        )
-    
     # Tips section
     with gr.Accordion("💡 Pro Tips for Best Results", open=False):
         gr.Markdown("""
@@ -277,6 +308,19 @@ with gr.Blocks(title=TITLE, theme=gr.themes.Base(), css="""
         - **Clarity**: Ensure text/contours are legible
         - **Completeness**: Include colorbars, titles, valid times
         
+        ### Using URLs:
+        - Supports direct links to JPG, PNG, GIF images
+        - Some weather sites may block direct access
+        - For best results, use official weather service URLs
+        - Alternative: Save image locally and upload
+        
+        ### Common Weather Data Sources:
+        - **SPC**: Convective outlooks, mesoanalysis
+        - **College of DuPage**: NEXRAD, model data
+        - **Tropical Tidbits**: Model analysis maps
+        - **GOES Imagery**: Satellite data
+        - **WPC**: Surface analysis, QPF
+        
         ### Analysis Tips by Data Type:
         
         **Model Output (GFS, HRRR, ECMWF, NAM):**
@@ -293,17 +337,6 @@ with gr.Blocks(title=TITLE, theme=gr.themes.Base(), css="""
         - Include timestamp and location
         - Specify interest in particular features
         - Ask about storm motion or development
-        
-        **Surface/Upper Air Charts:**
-        - Ensure contours and labels are clear
-        - Ask about specific features or patterns
-        - Request frontal analysis or jet stream discussion
-        
-        ### Parameter Settings:
-        - **Temperature 0.3-0.5**: Factual, consistent analysis
-        - **Temperature 0.6-0.8**: Balanced analysis with some interpretation
-        - **Temperature 0.9-1.0**: More speculative, exploring possibilities
-        - **Max Tokens**: Use 256-384 for quick analysis, 512-768 for detailed
         """)
     
     # Update mode description when mode changes
@@ -316,9 +349,15 @@ with gr.Blocks(title=TITLE, theme=gr.themes.Base(), css="""
     # Set up event handler
     analyze_btn.click(
         fn=analyze_weather_image,
-        inputs=[image_input, analysis_type, custom_prompt, analysis_mode, temperature, max_tokens, top_p],
+        inputs=[image_input, image_url_input, analysis_type, custom_prompt, analysis_mode, temperature, max_tokens, top_p],
         outputs=output
     )
+    
+    # Clear image upload when URL is entered
+    image_url_input.change(
+        lambda: None,
+        outputs=image_input
+    )
 
 # Launch the app
 if __name__ == "__main__":