Upload app.py

app.py

@@ -47,27 +47,27 @@ if radio_buttons == "GPU":
     ROPs = st.text_input("the number of ROPs")
     mem_type = st.radio("select the memory type", ("GDDR5", "GDDR5X", "GDDR6", "GDDR6X", "GDDR7"))
 
-    GPU_SPECS = gpu(
-        # memory and graphics related specs
-        transistors=int(transistors),  # million transistor
-        base_clock=int(base_clock),  # in MHz
-        mem_clock=int(mem_clock),  # in MHz
-        mem_size=int(mem_size),  # in GBs
-        shading_units=int(shading_units),
-        TMUs=int(TMUs),
-        ROPs=int(ROPs),
-
-        # memory type
-        mem_type_GDDR5=mem_type=="GDDR5",
-        mem_type_GDDR5X=mem_type=="GDDR5X",
-        mem_type_GDDR6=mem_type=="GDDR6",
-        mem_type_GDDR6X=mem_type=="GDDR6X",
-        mem_type_GDDR7=mem_type=="GDDR7"
-    )
     if transistors and base_clock and mem_clock and mem_size and shading_units and TMUs and ROPs and mem_type:
-        st.write("predictions with go here")
+        GPU_SPECS = gpu(
+            # memory and graphics related specs
+            transistors=int(transistors),  # million transistor
+            base_clock=int(base_clock),  # in MHz
+            mem_clock=int(mem_clock),  # in MHz
+            mem_size=int(mem_size),  # in GBs
+            shading_units=int(shading_units),
+            TMUs=int(TMUs),
+            ROPs=int(ROPs),
+
+            # memory type
+            mem_type_GDDR5=mem_type == "GDDR5",
+            mem_type_GDDR5X=mem_type == "GDDR5X",
+            mem_type_GDDR6=mem_type == "GDDR6",
+            mem_type_GDDR6X=mem_type == "GDDR6X",
+            mem_type_GDDR7=mem_type == "GDDR7"
+        )
     else:
-        st.write("you made a mistake")
+        st.write("you need to fill all the inputs")
+
 elif radio_buttons == "SSD":
     seq_read = st.text_input("the sequential read in MB/s")
     seq_write = st.text_input("the sequential write in MB/s")
@@ -76,27 +76,30 @@ elif radio_buttons == "SSD":
     interface = st.radio("give the SSD interface", ("PCIe 3.0 x4", "PCIe 4.0 x4", "PCIe 5.0 x4", "SATA 6GBps"))
     protocol = st.radio("give the protocol : ", ("AHCI", "NVMe", "NVMe 1.2", "NVMe 1.3", "NVMe 1.4", "NVMe 2.0"))
 
-    SSD_SPECS = ssd(
-        seq_read_write=int(seq_read) + int(seq_write),  # in MB/s (sequential read + sequential write)
-        endurance=int(endurance),  # in TBW
+    if seq_read and seq_write and endurance and type and interface and protocol:
+        SSD_SPECS = ssd(
+            seq_read_write=int(seq_read) + int(seq_write),  # in MB/s (sequential read + sequential write)
+            endurance=int(endurance),  # in TBW
 
-        # SSD types
-        type_MLC=type=="MLC",
-        type_QLC=type=="QLC",
-        type_SLC=type=="SLC",
-        type_TLC=type=="TLC",
+            # SSD types
+            type_MLC=type == "MLC",
+            type_QLC=type == "QLC",
+            type_SLC=type == "SLC",
+            type_TLC=type == "TLC",
 
-        # SSD interfaces
-        interface_PCIe_3_0_x4=interface=="PCIe 3.0 x4",
-        interface_PCIe_4_0_x4=interface=="PCIe 4.0 x4",
-        interface_PCIe_5_0_x4=interface=="PCIe 5.0 x4",
-        interface_SATA_6_Gbps=interface=="SATA 6GBps",
+            # SSD interfaces
+            interface_PCIe_3_0_x4=interface == "PCIe 3.0 x4",
+            interface_PCIe_4_0_x4=interface == "PCIe 4.0 x4",
+            interface_PCIe_5_0_x4=interface == "PCIe 5.0 x4",
+            interface_SATA_6_Gbps=interface == "SATA 6GBps",
 
-        # SSD protocols
-        protocol_AHCI=protocol=="AHCI",
-        protocol_NVMe=protocol=="NVMe",
-        protocol_NVMe_1_2=protocol=="NVMe 1.2",
-        protocol_NVMe_1_3=protocol=="NVMe 1.3",
-        protocol_NVMe_1_4=protocol=="NVMe 1.4",
-        protocol_NVMe_2_0=protocol=="NVMe 2.0"
-    )
+            # SSD protocols
+            protocol_AHCI=protocol == "AHCI",
+            protocol_NVMe=protocol == "NVMe",
+            protocol_NVMe_1_2=protocol == "NVMe 1.2",
+            protocol_NVMe_1_3=protocol == "NVMe 1.3",
+            protocol_NVMe_1_4=protocol == "NVMe 1.4",
+            protocol_NVMe_2_0=protocol == "NVMe 2.0"
+        )
+    else:
+        st.write("you need to fill all the inputs")