flaskAPI.py

flaskAPI.py

@@ -71,33 +71,5 @@ def check_vulnerabilities_api():
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
-def check_vulnerabilities_interface(solidity_file_url, file_id):
-    model_directory = 'models/'
-    device = 'cuda'
-    
-    try:
-        response = requests.get(solidity_file_url)
-        response.raise_for_status()
-
-        with tempfile.NamedTemporaryFile(delete=False, suffix=f"_{file_id}.sol") as temp_file:
-            temp_file.write(response.content)
-            temp_file.flush()
-            results = check_vulnerabilities(temp_file.name, model_directory, device)
-            os.remove(temp_file.name)
-        return results
-
-    except requests.exceptions.RequestException as e:
-        return {'error': f'Error fetching file: {e}'}
-    except Exception as e:
-        return {'error': str(e)}
-
-interface = gr.Interface(
-    fn=check_vulnerabilities_interface,
-    inputs=[gr.inputs.Textbox(label="Solidity File URL"), gr.inputs.Textbox(label="File ID")],
-    outputs="json",
-    title="Solidity Vulnerability Checker",
-    description="Enter the URL of a Solidity file and a file ID to check for vulnerabilities."
-)
-
 if __name__ == '__main__':
     app.run()