Update app.py

app.py

@@ -5,6 +5,7 @@ import random
 from datetime import datetime, timedelta
 import plotly.express as px
 import re
+import io
 
 api_key = os.getenv("CEREBRAS_API_KEY")
 url = "https://api.cerebras.ai/v1/chat/completions"
@@ -41,16 +42,31 @@ def generate_random_log():
         entries.append(f"{time} | {lat}N, {lon}W | Frequency: {freq} GHz | Signal Strength: {signal}% | Message: {message}")
     return "\n".join(entries)
 
-def analyze_log(log_text):
+def filter_log_by_frequency(log_text, min_freq, max_freq):
     if not log_text.strip():
-        return "Error: Please enter a log.", None, None
-    LOG_HISTORY.append(log_text)
+        return log_text
+    filtered = []
+    for line in log_text.split("\n"):
+        match = re.search(r"Frequency: (\d+\.\d) GHz", line)
+        if match:
+            freq = float(match.group(1))
+            if min_freq <= freq <= max_freq:
+                filtered.append(line)
+        else:
+            filtered.append(line)  # Keep non-frequency lines
+    return "\n".join(filtered) if filtered else "No logs in frequency range."
+
+def analyze_log(log_text, min_freq, max_freq):
+    filtered_log = filter_log_by_frequency(log_text, min_freq, max_freq)
+    if not filtered_log.strip() or filtered_log.startswith("No logs"):
+        return filtered_log, None, None, None
+    LOG_HISTORY.append(filtered_log)
     data = {
         "model": "llama-4-scout-17b-16e-instruct",
         "messages": [
             {
                 "role": "user",
-                "content": "Analyze this satellite radio log and summarize in bullet points. Include interference risk scores (0-100, low signal <70% = high risk >80, emergency = 90). Ensure frequencies in issues and details:\n- Issues (e.g., low signal, noise, interference with frequency, risk score)\n- High-priority messages (e.g., emergencies, warnings)\n- Key details (coordinates, times, frequencies, signal strengths)\nLog:\n" + log_text
+                "content": "Analyze this satellite radio log and summarize in bullet points. Include interference risk scores (0-100, low signal <70% = high risk >80, emergency = 90). Ensure frequencies in issues and details:\n- Issues (e.g., low signal, noise, interference with frequency, risk score)\n- High-priority messages (e.g., emergencies, warnings)\n- Key details (coordinates, times, frequencies, signal strengths)\nLog:\n" + filtered_log
             }
         ],
         "max_completion_tokens": 500,
@@ -72,7 +88,7 @@ def analyze_log(log_text):
         html += "</ul></div>"
         signals = []
         times = []
-        for line in log_text.split("\n"):
+        for line in filtered_log.split("\n"):
             if "Signal Strength" in line:
                 match = re.search(r"Signal Strength: (\d+)%", line)
                 if match:
@@ -81,9 +97,10 @@ def analyze_log(log_text):
                 if time_match:
                     times.append(time_match.group(1)[-5:])
         fig = px.line(x=times, y=signals, labels={"x": "Time", "y": "Signal (%)"}, title="Signal Trend") if signals and len(signals) == len(times) else None
-        return summary, html, fig
+        export_file = io.StringIO(summary)
+        return summary, html, fig, gr.File(value=export_file, file_name="summary.txt", visible=True)
     except Exception as e:
-        return f"Error: API call failed - {str(e)}", None, None
+        return f"Error: API call failed - {str(e)}", None, None, None
 
 def generate_alert(log_text):
     if not log_text.strip():
@@ -151,6 +168,7 @@ with gr.Blocks(css=css) as interface:
     gr.Markdown("# Satellite Signal Log Analyzer", elem_classes="header")
     gr.Markdown("Analyze logs for issues, alerts, and trends.", elem_classes="subheader")
     log_input = gr.Textbox(lines=5, show_label=False, placeholder="Enter or generate a log...")
+    freq_slider = gr.Slider(minimum=14.0, maximum=15.0, step=0.1, value=[14.0, 15.0], label="Frequency Range (GHz)")
     with gr.Row():
         sample_button = gr.Button("Sample Log")
         random_button = gr.Button("Random Log")
@@ -159,15 +177,18 @@ with gr.Blocks(css=css) as interface:
         analyze_button = gr.Button("Analyze")
         alert_button = gr.Button("Alert")
         compare_button = gr.Button("Compare Logs")
+        export_button = gr.Button("Export")
     output = gr.HTML(show_label=False)
     plot_output = gr.Plot(show_label=False)
     alert_output = gr.HTML(show_label=False)
     compare_output = gr.HTML(show_label=False)
+    export_output = gr.File(show_label=False, visible=False)
     sample_button.click(fn=load_sample_log, outputs=log_input)
     random_button.click(fn=generate_random_log, outputs=log_input)
     clear_button.click(fn=clear_log, outputs=log_input)
-    analyze_button.click(fn=analyze_log, inputs=log_input, outputs=[output, output, plot_output])
+    analyze_button.click(fn=analyze_log, inputs=[log_input, freq_slider], outputs=[output, output, plot_output, export_output])
     alert_button.click(fn=generate_alert, inputs=log_input, outputs=alert_output)
     compare_button.click(fn=compare_logs, outputs=[compare_output, compare_output])
+    export_button.click(fn=lambda: None, outputs=export_output)
 
 interface.launch()